linfan 2 years ago
parent
commit
965bdc3b1a
1 changed files with 4 additions and 2 deletions
  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 = []
     video_items = []
     for i in range(len(video_scores)):
     for i in range(len(video_scores)):
         try:
         try:
-            print(video_scores[i])
+            #print(video_scores[i])
             if video_scores[i] is None:
             if video_scores[i] is None:
                 video_items.append((videoIds[i], 0.0))
                 video_items.append((videoIds[i], 0.0))
             else:
             else:
                 video_score_str = json.load(video_scores[i])
                 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]))
                 video_items.append((videoIds[i], video_score_str[0]))
         except Exception:
         except Exception:
             video_items.append((videoIds[i], 0.0))
             video_items.append((videoIds[i], 0.0))
     sort_items = sorted(video_items, key=lambda k: k[1], reverse=True)
     sort_items = sorted(video_items, key=lambda k: k[1], reverse=True)
+    print("sort_items:", sort_items)
     rov_recall_rank = sort_items
     rov_recall_rank = sort_items
     fast_flow_recall_rank = []
     fast_flow_recall_rank = []
     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_result.append(recall_item)
                     rank_set.add(recall_item[0])
                     rank_set.add(recall_item[0])
         i += 1
         i += 1
+    print("rank_result:", rank_result[:size])
     return rank_result[:size]
     return rank_result[:size]