linfan il y a 2 ans
Parent
commit
84840f3da5
2 fichiers modifiés avec 5 ajouts et 5 suppressions
  1. 4 4
      recommend.py
  2. 1 1
      video_rank.py

+ 4 - 4
recommend.py

@@ -331,7 +331,8 @@ def video_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type, al
         #print("test config:",test_config)
         rank_result, flow_result = video_sanke_rank(data=data, size=size, top_K=top_K, flow_pool_P=float(flow_pool_P), ab_Code=ab_code, exp_config=test_config, flowConfig=flow_config)
         result['flow_num'] = flow_result
-        result['rank_num'] = len(rank_result)
+        if rank_result:
+            result['rank_num'] = len(rank_result)
     else:
         rank_result = video_rank(data=data, size=size, top_K=top_K, flow_pool_P=float(flow_pool_P))
 
@@ -580,10 +581,9 @@ def video_old_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
     elif ab_code == 60057:
         flow_config = pool_recall.get_flow_exp_6_config()
     rank_result, flow_num = video_new_rank2(data=data, size=size, top_K=top_K, flow_pool_P=float(flow_pool_P), ab_code=ab_code, flowConfig=flow_config)
-    print("result:", len(rank_result))
-
     result['rankResult'] = rank_result
-    result['rank_num'] = len(rank_result)
+    if rank_result:
+        result['rank_num'] = len(rank_result)
     result['flow_num'] = flow_num
     result['rankTime'] = (time.time() - start_rank) * 1000
 

+ 1 - 1
video_rank.py

@@ -822,7 +822,7 @@ def video_sanke_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=None
         all_recall_rank = rov_recall_rank + flow_recall_rank
         if flow_num > 0:
             rank_result.extend(all_recall_rank[:size - top_K])
-            print("flow_num:", flow_num, len(rank_result))
+            #print("flow_num:", flow_num, len(rank_result))
             return rank_result[:size], flow_num
         else:
             # 按概率 p 及score排序获取 size - k 个视频