Browse Source

update rank

linfan 1 year ago
parent
commit
4228d1dfb8
2 changed files with 3 additions and 2 deletions
  1. 2 2
      rank_service.py
  2. 1 0
      video_rank.py

+ 2 - 2
rank_service.py

@@ -245,9 +245,9 @@ def get_tf_serving_sores(feature_dict):
     }
     request_data_dict= {}
     request_data_dict["inputs"] = inputs_data
-    #print(request_data_dict)
+    print(request_data_dict)
     request_data= json.dumps(request_data_dict)
-    #print(request_data)
+    print(request_data)
     # 调用http接口
     result = request_post_data(config_.TF_SERVING_URL,request_data, timeout=(0.1, 1))
     #print("result:", result)

+ 1 - 0
video_rank.py

@@ -640,6 +640,7 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, mid, exp_config=Non
         #print("env_dict:", env_dict)
         feature_dict = get_featurs(mid, data, size, top_K, flow_pool_P, rec_recall_vid_list,env_dict, video_static_info, video_hour_static_info)
         score_result = get_tf_serving_sores(feature_dict)
+        print("score_result:", score_result)
         if video_scores and len(video_scores)>0  and rec_recall_item_list and score_result and len(score_result) > 0\
                 and len(score_result) == len(rec_recall_item_list) and len(video_scores)== len(score_result):
             for i in range(len(score_result)):