liqian 1 anno fa
parent
commit
3a1e2b2890
3 ha cambiato i file con 9 aggiunte e 4 eliminazioni
  1. 5 2
      recommend.py
  2. 3 1
      video_rank.py
  3. 1 1
      video_recall.py

+ 5 - 2
recommend.py

@@ -1,3 +1,4 @@
+import copy
 import json
 import random
 import time
@@ -304,6 +305,7 @@ def video_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type, al
     result['recallTime'] = (time.time() - start_recall) * 1000
 
     # add_flow_pool_recall_log
+    recall_result_list = copy.deepcopy(recall_result_list)
     flow_pool_recall_process = {}
 
     # ####### 排序
@@ -652,9 +654,9 @@ def video_old_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
                 recall_result_list[0] = rov_pool_recall
     result['recallResult'] = recall_result_list
     result['recallTime'] = (time.time() - start_recall) * 1000
-    #print("recall:", recall_result_list)
 
     # add_flow_pool_recall_log
+    recall_result_list = copy.deepcopy(recall_result_list)
     flow_pool_recall_process = {}
 
     # ####### 排序
@@ -694,7 +696,7 @@ def video_old_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
                 # 'flow_pool_recall': recall_result_list[1]
             }
             # add_flow_pool_recall_log
-            flow_pool_recall_process = recall_result_list[1][1]
+            flow_pool_recall_process = recall_result_list[1][1].copy()
         else:
             data = {
                 'rov_pool_recall': recall_result_list[0],
@@ -741,6 +743,7 @@ def video_old_recommend(request_id, mid, uid, size, top_K, flow_pool_P, app_type
     # )
 
     # add_flow_pool_recall_log
+
     rank_result, flow_num, flow_pool_recall_process = video_new_rank3(
         data=data, size=size, top_K=top_K, flow_pool_P=float(flow_pool_P), rank_key_prefix=rank_key_prefix,
         flow_pool_recall_process=flow_pool_recall_process

+ 3 - 1
video_rank.py

@@ -1,3 +1,4 @@
+import copy
 import json
 import random
 import numpy
@@ -841,7 +842,8 @@ def video_new_rank3(data, size, top_K, flow_pool_P, rank_key_prefix='rank:score1
     rank_result = []
 
     # add_flow_pool_recall_log
-    flow_pool_recall_process['recall_duplicate_res'] = {'rov_recall_rank': rov_recall_rank, 'flow_recall_rank': flow_recall_rank}
+    flow_pool_recall_process['recall_duplicate_res'] = {'rov_recall_rank': rov_recall_rank,
+                                                        'flow_recall_rank': copy.deepcopy(flow_recall_rank)}
 
     # 从ROV召回池中获取top k
     if len(rov_recall_rank) > 0:

+ 1 - 1
video_recall.py

@@ -877,7 +877,7 @@ class PoolRecall(object):
                                                               start=idx, end=idx + get_size - 1,
                                                               with_scores=True)
             # add_flow_pool_recall_log
-            print(data)
+            # print(data)
             flow_pool_recall_process['initial_data'] = data
 
             # et_get = time.time()