linfan 1 年之前
父節點
當前提交
e8cd9d7aa0
共有 2 個文件被更改,包括 16 次插入11 次删除
  1. 14 11
      recommend.py
  2. 2 0
      video_recall.py

+ 14 - 11
recommend.py

@@ -393,17 +393,20 @@ def video_old_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
             region_recall = recall_result_list[0]
             sim_recall = recall_result_list[3]
             now_video_ids = set('')
-            for video in region_recall:
-                video_id = video.get('videoId')
-                if video_id not in now_video_ids:
-                    rov_pool_recall.append(video)
-                    now_video_ids.add(video_id)
-            for video in sim_recall:
-                video_id = video.get('videoId')
-                if video_id not in now_video_ids:
-                    rov_pool_recall.append(video)
-                    now_video_ids.add(video_id)
-            recall_result_list[0] = rov_pool_recall
+            if len(region_recall)>0:
+                for video in region_recall:
+                    video_id = video.get('videoId')
+                    if video_id not in now_video_ids:
+                        rov_pool_recall.append(video)
+                        now_video_ids.add(video_id)
+            if len(sim_recall)>0:
+                for video in sim_recall:
+                    video_id = video.get('videoId')
+                    if video_id not in now_video_ids:
+                        rov_pool_recall.append(video)
+                        now_video_ids.add(video_id)
+            if len(rov_pool_recall)>0:
+                recall_result_list[0] = rov_pool_recall
 
     result['recallResult'] = recall_result_list
     result['recallTime'] = (time.time() - start_recall) * 1000

+ 2 - 0
video_recall.py

@@ -2142,6 +2142,8 @@ class PoolRecall(object):
                      'rovScore': per_item[1], 'pushFrom': config_.PUSH_FROM['sim_hot_vid_recall'],
                      'abCode': self.ab_code}
 
+        if len(video_ids)<=0:
+            return  recall_result
         video_ids = video_ids[:50]
         print(video_ids)
         filter_ = FilterVideos(request_id=self.request_id,