Browse Source

update test

linfan 1 year ago
parent
commit
3d2e18e7e9
1 changed files with 7 additions and 2 deletions
  1. 7 2
      video_rank.py

+ 7 - 2
video_rank.py

@@ -921,11 +921,16 @@ def video_sank_pos_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=N
     sim_recall = []
     u2u2i_recall = []
     return_video_recall = []
+    #print("")
     if ab_Code==60058:
         if len(data['u2i_recall'])>0:
-            u2i_recall = sorted(data['u2i_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
+            recall_dict['u2i_recall'] = data['u2i_recall']
+        else:
+            recall_dict['u2i_recall'] = u2i_recall
         if len(data['u2i_play_recall']) > 0:
-            u2i_play_recall = sorted(data['u2i_play_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
+            recall_dict['u2i_play_recall'] = data['u2i_play_recall']
+        else:
+            recall_dict['u2i_play_recall'] = u2i_play_recall
     elif ab_Code==60059:
         if len(data['w2v_recall'])>0:
             recall_dict['w2v_recall'] = data['w2v_recall']