linfan 1 year ago
parent
commit
692fb27c9f
2 changed files with 7 additions and 8 deletions
  1. 1 2
      recommend.py
  2. 6 6
      video_rank.py

+ 1 - 2
recommend.py

@@ -463,8 +463,6 @@ 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,
                            app_type=app_type, mid=mid, uid=uid, video_ids=list(recall_dict.keys()))
@@ -501,6 +499,7 @@ def new_video_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
             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

+ 6 - 6
video_rank.py

@@ -167,9 +167,9 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
     vidKeys = []
     for vid in videoIds:
         vidKeys.append("k_p:"+str(vid))
-    print("vidKeys:", vidKeys)
+    #print("vidKeys:", vidKeys)
     video_scores = redisObj.get_batch_key(vidKeys)
-    print(video_scores)
+    #print(video_scores)
     video_items = []
     for i in range(len(video_scores)):
         try:
@@ -178,12 +178,12 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
                 video_items.append((videoIds[i], 0.0))
             else:
                 video_score_str = json.loads(video_scores[i])
-                print("video_score_str:",video_score_str)
+                #print("video_score_str:",video_score_str)
                 video_items.append((videoIds[i], video_score_str[0]))
         except Exception:
             video_items.append((videoIds[i], 0.0))
     sort_items = sorted(video_items, key=lambda k: k[1], reverse=True)
-    print("sort_items:", sort_items)
+    #print("sort_items:", sort_items)
     rov_recall_rank = sort_items
     fast_flow_recall_rank = []
     flow_recall_rank = []
@@ -207,7 +207,7 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
 
     for rank_item in rank_result:
         rank_set.add(rank_item[0])
-
+    print("rank_result:", rank_result)
     # 按概率 p 及score排序获取 size - k 个视频, 第4个位置按概率取流量池
     i = 0
     while i < size - top_K:
@@ -229,7 +229,7 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
                     rank_result.append(recall_item)
                     rank_set.add(recall_item[0])
         i += 1
-    print("rank_result:", rank_result[:size])
+    print("rank_result:", rank_result)
     return rank_result[:size]