瀏覽代碼

update rank

linfan 1 年之前
父節點
當前提交
022f8ff8b6
共有 2 個文件被更改,包括 8 次插入10 次删除
  1. 2 2
      rank_service.py
  2. 6 8
      utils.py

+ 2 - 2
rank_service.py

@@ -180,9 +180,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))
 

+ 6 - 8
utils.py

@@ -44,17 +44,17 @@ def request_post(request_url, request_data, timeout):
     """
     try:
         headers = {"Connection": "close"}
-        print(request_url)
-        print(headers)
+        #print(request_url)
+        #print(headers)
         response = requests.post(url=request_url, json=request_data, timeout=timeout, headers=headers)
-        print("response:", response)
+        #print("response:", response)
         if response.status_code == 200:
             res_data = json.loads(response.text)
             return res_data
         else:
             return None
     except Exception as e:
-        print(e)
+        #print(e)
         log_.error('url: {}, exception: {}, traceback: {}'.format(request_url, e, traceback.format_exc()))
         return None
 
@@ -69,17 +69,15 @@ def request_post_data(request_url, request_data, timeout):
     """
     try:
         headers = {'content-type': 'application/json'}
-        print(request_url)
-        print(headers)
         response = requests.post(url=request_url, data=request_data, timeout=timeout, headers=headers)
-        print("response:", response)
+        #print("response:", response)
         if response.status_code == 200:
             res_data = json.loads(response.text)
             return res_data['outputs']
         else:
             return None
     except Exception as e:
-        print(e)
+        #print(e)
         log_.error('url: {}, exception: {}, traceback: {}'.format(request_url, e, traceback.format_exc()))
         return None