瀏覽代碼

update video rank

linfan 1 年之前
父節點
當前提交
ac6c4ee64c
共有 1 個文件被更改,包括 4 次插入3 次删除
  1. 4 3
      video_rank.py

+ 4 - 3
video_rank.py

@@ -973,17 +973,17 @@ def video_sank_pos_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=N
     if exp_config and 'recall_num_limit' in exp_config:
         recall_num_limit_dict = exp_config['recall_num_limit']
     exp_recall_dict = {}
-    index_pos = 0
+   #index_pos = 0
     for j in range(3):
         if len(rov_recall_rank)>12:
             break
         # choose pos
         for recall_pos_config in recall_list:
             rand_num = random.random()
-            index_pos +=1
+            index_pos = 0
             # choose pos recall
             for per_recall_item in recall_pos_config:
-                if index_pos <= len(rov_recall_rank):
+                if index_pos == 1:
                     break
                 per_recall_name = per_recall_item[0]
                 per_recall_freq = per_recall_item[1]
@@ -1005,6 +1005,7 @@ def video_sank_pos_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=N
                             exp_recall_dict[per_recall_name] +=1
                         else:
                             exp_recall_dict[per_recall_name] = 1
+                        index_pos = 1
                         break
     #print("rov_recall_rank:", rov_recall_rank)
     if len(rov_recall_rank)<4: