linfan 1 year ago
parent
commit
16709aad1a
1 changed files with 3 additions and 3 deletions
  1. 3 3
      video_rank.py

+ 3 - 3
video_rank.py

@@ -676,7 +676,7 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code):
         i += 1
     return rank_result[:size]
 
-def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code='', test_config=None):
+def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=None):
     """
     视频分发排序
     :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
@@ -726,8 +726,8 @@ def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code='', test_config=Non
     recall_list = [('rov_recall_region_h',1, 1),('hot_recall',0.5,1), ('w2v_recall',0.5,1),
                    ('rov_recall_region_h',0.5, 1),('rov_recall_region_24h',1,1),
                    ('rov_recall_24h',1,1), ('rov_recall_24h_dup',0.5,1)]
-    if not test_config:
-        recall_list = test_config
+    if exp_config is not  None:
+        recall_list = exp_config
     print("recall_list:", recall_list)
     rov_recall_rank = []
     select_ids = set('')