linfan 1 gadu atpakaļ
vecāks
revīzija
583224e87b
1 mainītis faili ar 9 papildinājumiem un 3 dzēšanām
  1. 9 3
      recommend.py

+ 9 - 3
recommend.py

@@ -459,6 +459,7 @@ def new_video_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
         else:
             recall_name = recall_dict[vId] + "," + recall_name
             recall_dict[vId] = recall_name
+        
     #all_flow_set = set.union(fast_flow_set, flow_flow_set)
     #3. filter video, 先过预曝光
     filter_ = FilterVideos(request_id=request_id,
@@ -489,9 +490,14 @@ def new_video_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
     rank_result= []
     if ab_code==60048 or ab_code==60049:
         rank_ids = video_new_rank(videoIds=all_recall_list,fast_flow_set=fast_flow_set, flow_set=flow_flow_set,size=size, top_K=top_K, flow_pool_P=float(flow_pool_P))
-        for rank_id in rank_ids:
-            if rank_id in recall_dict:
-                rank_result.append(recall_dict.get(rank_id))
+        for rank_item in rank_ids:
+            rank_id = rank_item[0]
+            rank_score = rank_item[1]
+            pushFrom = recall_dict.get(rank_id, '')
+            rank_result.append({'videoId': rank_id, 'flowPool': '',
+                     'rovScore': rank_score, 'pushFrom': pushFrom,
+                     'abCode': ab_code})
+        print("rank_result:", rank_result)
     else:
         all_dup_recall_result = region_h_recall+region_day_recall+select_day_recall+no_selected_recall
         rank_result = refactor_video_rank(rov_recall_rank=all_dup_recall_result,fast_flow_set=fast_flow_set, flow_set=flow_flow_set, size=size, top_K=top_K, flow_pool_P=float(flow_pool_P))