liqian 3 years ago
parent
commit
c4b05c0531
2 changed files with 2 additions and 1 deletions
  1. 1 1
      config.py
  2. 1 0
      video_recall.py

+ 1 - 1
config.py

@@ -3,7 +3,7 @@ class BaseConfig(object):
     AB_CODE = 10000
     # category id mapping
     CATEGORY = {
-        'recommend': [0],  # 推荐
+        'recommend': [55],  # 推荐
         'other': [1, 2, 3]  # 其他类别
     }
     # 前k个从ROV召回池中获取视频

+ 1 - 0
video_recall.py

@@ -87,6 +87,7 @@ class PoolRecall(object):
         freq = 0
         idx = 0
         while len(flow_pool_recall_result) < size:
+            freq += 1
             # 获取数据
             st_get = time.time()
             data = self.redis_helper.get_data_zset_with_index(key_name=flow_pool_key,