소스 검색

add video_rank

linfan 2 년 전
부모
커밋
ea3293371a
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      video_rank.py

+ 1 - 1
video_rank.py

@@ -965,7 +965,7 @@ def video_sank_pos_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=N
             for per_recall_item in recall_pos_config:
                 per_recall_name = per_recall_item[0]
                 per_recall_freq = per_recall_item[1]
-                if rand_num < per_recall_freq and per_recall_name in recall_dict
+                if rand_num < per_recall_freq and per_recall_name in recall_dict:
                     per_recall = recall_dict[per_recall_name]
                     for recall_item in per_recall:
                         vid = recall_item['videoId']