Browse Source

update rank

linfan 1 year ago
parent
commit
4dee8c1d7a
3 changed files with 3 additions and 4 deletions
  1. 0 1
      rank_service.py
  2. 1 2
      utils.py
  3. 2 1
      video_rank.py

+ 0 - 1
rank_service.py

@@ -191,6 +191,5 @@ def get_tf_serving_sores(feature_dict):
         print("result is None")
         log_.info('call tf serving error,types: {}')
         return []
-
     print(result)
     return result

+ 1 - 2
utils.py

@@ -75,8 +75,7 @@ def request_post_data(request_url, request_data, timeout):
         print("response:", response)
         if response.status_code == 200:
             res_data = json.loads(response.text)
-            print(res_data)
-            return res_data
+            return res_data['outputs']
         else:
             return None
     except Exception as e:

+ 2 - 1
video_rank.py

@@ -633,7 +633,8 @@ def video_new_rank2(data, size, top_K, flow_pool_P, ab_code, exp_config=None):
         score_result = get_tf_serving_sores(feature_dict)
         if score_result and len(score_result) > 0 and len(score_result) == len(recall_list):
             for i in range(len(score_result)):
-                recall_list[i]['sort_score'] = score_result[i]
+                recall_list[i]['sort_score'] = score_result[i][0]
+                print(score_result[i][0])
             rov_recall_rank = sorted(recall_list, key=lambda k: k.get('sort_score', 0), reverse=True)
     else:
         redisObj = RedisHelper()