浏览代码

update video_rank

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

+ 2 - 2
video_rank.py

@@ -184,7 +184,7 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
         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)
+    #print("sort_items:", sort_items)
     rov_recall_rank = sort_items
     fast_flow_recall_rank = []
     flow_recall_rank = []
@@ -214,7 +214,7 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
     left_quato = size - top_K
     j = 0
     jj = 0
-    while i < left_quato and j<len(all_flow_recall_rank) and jj<len(rov_recall_rank):
+    while i < left_quato and (j<len(all_flow_recall_rank) or jj<len(rov_recall_rank)):
         # 随机生成[0, 1)浮点数
         rand = random.random()
         # log_.info('rand: {}'.format(rand))