liqian %!s(int64=2) %!d(string=hai) anos
pai
achega
5b617c8516
Modificáronse 2 ficheiros con 3 adicións e 2 borrados
  1. 1 0
      config.py
  2. 2 2
      videos_similarity.py

+ 1 - 0
config.py

@@ -261,6 +261,7 @@ class BaseConfig(object):
     RECALL_POSITION1_KEY_NAME = 'com.weiqu.video.recall.hot.apptype.h.item.score.position.1'
     RECALL_POSITION2_KEY_NAME = 'com.weiqu.video.recall.hot.apptype.h.item.score.position.2'
 
+    SIM_N_19 = 4
     # 最惊奇电影类视频相关推荐列表存放 redis key前缀,完整格式: com.weiqu.movie.relevant.list.item.{videoId}
     MOVIE_RELEVANT_LIST_KEY_NAME_PREFIX = 'com.weiqu.movie.relevant.list.item.'
 

+ 2 - 2
videos_similarity.py

@@ -131,11 +131,11 @@ def similarity_rank(movie_videos, sim_videos):
                 item_sim = {'top_video_id': video_id, 'title': title, 'vid': vid, 'title1': title1, 'dist': dist}
                 item_sim_list.append(item_sim)
         item_sim_list.sort(key=lambda x: x['dist'], reverse=True)
-        sim_result.extend(item_sim_list[:4])
+        sim_result.extend(item_sim_list[:config_.SIM_N_19])
         # to_redis
         key_name = f"{config_.MOVIE_RELEVANT_LIST_KEY_NAME_PREFIX}{video_id}"
         relevant_data = dict()
-        for item in item_sim_list:
+        for item in item_sim_list[:config_.SIM_N_19]:
             relevant_data[item['vid']] = item['dist']
         if redis_helper.key_exists(key_name=key_name):
             redis_helper.del_keys(key_name=key_name)