linfan 1 éve
szülő
commit
714baece02
1 módosított fájl, 3 hozzáadás és 2 törlés
  1. 3 2
      video_rank.py

+ 3 - 2
video_rank.py

@@ -160,8 +160,9 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
         :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
         :return: rank_result
         """
+    add_flow_set = set('')
     if not videoIds or len(videoIds)==0:
-        return []
+        return [], add_flow_set
 
     redisObj = RedisHelper()
     vidKeys = []
@@ -211,7 +212,7 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
     # 按概率 p 及score排序获取 size - k 个视频, 第4个位置按概率取流量池
     i = 0
     left_quato = size - top_K
-    add_flow_set = set('')
+
     while i < left_quato:
         # 随机生成[0, 1)浮点数
         rand = random.random()