linfan 1 year ago
parent
commit
dec93fa302
1 changed files with 8 additions and 2 deletions
  1. 8 2
      recommend.py

+ 8 - 2
recommend.py

@@ -291,9 +291,15 @@ def video_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type, al
             }
 
     if ab_code == 60058:
-        data['hot_recall'] = recall_result_list[3]
+        if len(recall_result_list)>=4:
+            data['hot_recall'] = recall_result_list[3]
+        else:
+            data['hot_recall'] = []
     elif ab_code == 60059:
-        data['w2v_recall'] = recall_result_list[3]
+        if len(recall_result_list)>=4:
+            data['w2v_recall'] = recall_result_list[3]
+        else:
+            data['hot_recall'] = []
     #if ab_code=="ab_new_test":
     #    rank_result = video_new_rank(data=data, size=size, top_K=top_K, flow_pool_P=float(flow_pool_P))
     #else: