Explorar o código

update video_rank

linfan hai 1 ano
pai
achega
d23f64e18e
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  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))