ソースを参照

updaet video_rank

linfan 1 年間 前
コミット
9736c408c4
1 ファイル変更2 行追加0 行削除
  1. 2 0
      video_rank.py

+ 2 - 0
video_rank.py

@@ -628,6 +628,7 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, mid, exp_config=Non
     recall_list = []
     rov_recall_rank = data['rov_pool_recall']
     #call rank service
+    #flag_call_service = 0
     if ab_code == 60066:
         feature_dict, recall_list = get_featurs(mid, data, size, top_K, flow_pool_P, env_dict)
         score_result = get_tf_serving_sores(feature_dict)
@@ -635,6 +636,7 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, mid, exp_config=Non
             for i in range(len(score_result)):
                 recall_list[i]['sort_score'] = score_result[i][0]
                 #print(score_result[i][0])
+                recall_list[i]['flag_call_service'] = 1
             rov_recall_rank = sorted(recall_list, key=lambda k: k.get('sort_score', 0), reverse=True)
     else:
         redisObj = RedisHelper()