Parcourir la source

udapte video_rank.py

linfan il y a 1 an
Parent
commit
fecdad354f
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      video_rank.py

+ 1 - 1
video_rank.py

@@ -813,7 +813,7 @@ def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=None
         rank_result.extend(flow_recall_rank[:top_K])
         flow_recall_rank = flow_recall_rank[top_K:]
     flow_num = 0
-    if len(rov_recall_rank) > 0:
+    if flowConfig == "1" and len(rov_recall_rank) > 0:
         rank_result.extend(rov_recall_rank[:top_K])
         for recall_item in rank_result:
             flow_recall_name = recall_item.get("flowPool", '')