linfan 1 year ago
parent
commit
4a24b88bb1
1 changed files with 5 additions and 3 deletions
  1. 5 3
      video_rank.py

+ 5 - 3
video_rank.py

@@ -720,12 +720,14 @@ def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code=''):
     if ab_Code==60058:
         if len(data['hot_recall'])>0:
             hot_recall = sorted(data['hot_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
-            recall_dict['hot_recall'] = hot_recall
+        recall_dict['hot_recall'] = hot_recall
     elif ab_Code==60059:
         #print(data['w2v_recall'])
         if len(data['w2v_recall'])>0:
-           recall_dict['w2v_recall'] = data['w2v_recall']
-           print("recall_dict['w2v_recall']", recall_dict['w2v_recall'])
+            recall_dict['w2v_recall'] = data['w2v_recall']
+        else:
+            recall_dict['w2v_recall'] = w2v_recall
+        print("recall_dict['w2v_recall']", recall_dict['w2v_recall'])
 
     recall_list = [('rov_recall_region_h',1, 1),('hot_recall',0.5,1), ('w2v_recall',0.5,1),
                    ('rov_recall_region_h',0.5, 1),('rov_recall_region_24h',1,1),