linfan 1 年之前
父節點
當前提交
dec93fa302
共有 1 個文件被更改,包括 8 次插入2 次删除
  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: