linfan 1 year ago
parent
commit
cda3c23fb3
1 changed files with 27 additions and 29 deletions
  1. 27 29
      video_rank.py

+ 27 - 29
video_rank.py

@@ -656,36 +656,34 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, flowConfig=None):
         # 按概率 p 及score排序获取 size - k 个视频
     flow_num = 0
     if flowConfig == "1" and len(rov_recall_rank) > 0:
-        if flowConfig == "1" and len(rov_recall_rank) > 0:
-            for recall_item in rank_result:
-                flow_recall_name = recall_item.get("flowPool", '')
-                flow_num = flow_num + 1
-            all_recall_rank = rov_recall_rank + flow_recall_rank
-            if flow_num > 0:
-                rank_result.extend(all_recall_rank[:size - top_K])
-                return  rank_result, flow_num
-            else:
-                i = 0
-                while i < size - top_K:
-                    # 随机生成[0, 1)浮点数
-                    rand = random.random()
-                    # log_.info('rand: {}'.format(rand))
-                    if rand < flow_pool_P:
-                        if flow_recall_rank:
-                            rank_result.append(flow_recall_rank[0])
-                            flow_recall_rank.remove(flow_recall_rank[0])
-                        else:
-                            rank_result.extend(rov_recall_rank[:size - top_K - i])
-                            return rank_result[:size], flow_num
+        for recall_item in rank_result:
+            flow_recall_name = recall_item.get("flowPool", '')
+            flow_num = flow_num + 1
+        all_recall_rank = rov_recall_rank + flow_recall_rank
+        if flow_num > 0:
+            rank_result.extend(all_recall_rank[:size - top_K])
+            return rank_result, flow_num
+        else:
+            i = 0
+            while i < size - top_K:
+                # 随机生成[0, 1)浮点数
+                rand = random.random()
+                # log_.info('rand: {}'.format(rand))
+                if rand < flow_pool_P:
+                    if flow_recall_rank:
+                        rank_result.append(flow_recall_rank[0])
+                        flow_recall_rank.remove(flow_recall_rank[0])
                     else:
-                        if rov_recall_rank:
-                            rank_result.append(rov_recall_rank[0])
-                            rov_recall_rank.remove(rov_recall_rank[0])
-                        else:
-                            rank_result.extend(flow_recall_rank[:size - top_K - i])
-                            return rank_result[:size], flow_num
-                    i += 1
-
+                        rank_result.extend(rov_recall_rank[:size - top_K - i])
+                        return rank_result[:size], flow_num
+                else:
+                    if rov_recall_rank:
+                        rank_result.append(rov_recall_rank[0])
+                        rov_recall_rank.remove(rov_recall_rank[0])
+                    else:
+                        rank_result.extend(flow_recall_rank[:size - top_K - i])
+                        return rank_result[:size], flow_num
+                i += 1
     else:
         i = 0
         while i < size - top_K: