liqian 3 years ago
parent
commit
71986c2edc
1 changed files with 3 additions and 0 deletions
  1. 3 0
      video_rank.py

+ 3 - 0
video_rank.py

@@ -27,6 +27,8 @@ def video_rank(data, size):
     flow_recall_rank = sorted(data['flow_pool_recall'], key=lambda k: (k.get('rovScore'), 0), reverse=True)
     # 对各路召回的视频进行去重
     rov_recall_rank, flow_recall_rank = remove_duplicate(rov_recall=rov_recall_rank, flow_recall=flow_recall_rank)
+    log_.info('remove_duplicate finished! rov_recall_rank = {}, flow_recall_rank = {}'.format(
+        rov_recall_rank, flow_recall_rank))
     # 从ROV召回池中获取top k
     if len(rov_recall_rank) > 0:
         rank_result = rov_recall_rank[:config_.K]
@@ -40,6 +42,7 @@ def video_rank(data, size):
     while i < size - config_.K:
         # 随机生成[0, 1)浮点数
         rand = random.random()
+        log_.info('rand: {}'.format(rand))
         if rand < config_.P:
             if flow_recall_rank:
                 rank_result.append(flow_recall_rank[0])