linfan 1 年之前
父节点
当前提交
965bdc3b1a
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      video_rank.py

+ 4 - 2
video_rank.py

@@ -173,16 +173,17 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
     video_items = []
     for i in range(len(video_scores)):
         try:
-            print(video_scores[i])
+            #print(video_scores[i])
             if video_scores[i] is None:
                 video_items.append((videoIds[i], 0.0))
             else:
                 video_score_str = json.load(video_scores[i])
-                print("video_score_str:",video_score_str)
+                #print("video_score_str:",video_score_str)
                 video_items.append((videoIds[i], video_score_str[0]))
         except Exception:
             video_items.append((videoIds[i], 0.0))
     sort_items = sorted(video_items, key=lambda k: k[1], reverse=True)
+    print("sort_items:", sort_items)
     rov_recall_rank = sort_items
     fast_flow_recall_rank = []
     flow_recall_rank = []
@@ -228,6 +229,7 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
                     rank_result.append(recall_item)
                     rank_set.add(recall_item[0])
         i += 1
+    print("rank_result:", rank_result[:size])
     return rank_result[:size]