Quellcode durchsuchen

update rank_service

linfan vor 1 Jahr
Ursprung
Commit
b722dccaf3
1 geänderte Dateien mit 8 neuen und 2 gelöschten Zeilen
  1. 8 2
      rank_service.py

+ 8 - 2
rank_service.py

@@ -21,6 +21,7 @@ def get_featurs(mid, data, size, top_K, flow_pool_P, env_dict=None):
     machineinfo_model ='M2006C3LC'
     city_code = -1
     province_code = -1
+    relevant_video_id = -1
     if env_dict and len(env_dict)>0:
         apptype = env_dict.get('app_type',4)
         pagesource = env_dict.get('pagesource', '')
@@ -32,6 +33,7 @@ def get_featurs(mid, data, size, top_K, flow_pool_P, env_dict=None):
         sencetype = env_dict.get('sencetype', '')
         city_code = env_dict.get('city_code', -1)
         province_code = env_dict.get('province_code', -1)
+        relevant_video_id = env_dict.get('relevant_video_id', -1)
         if city_code and city_code == "":
             city_code = -1
         if province_code and province_code == "":
@@ -66,6 +68,7 @@ def get_featurs(mid, data, size, top_K, flow_pool_P, env_dict=None):
     day_view_users_list = []
     day_share_users = []
     recommendsource_list = []
+    relevant_video_list = []
     recall_list = env_dict.get('recall_list', [])
     city_list = []
     province_list = []
@@ -83,6 +86,7 @@ def get_featurs(mid, data, size, top_K, flow_pool_P, env_dict=None):
             sencetype_list.append(sencetype)
             city_list.append(city_code)
             province_list.append(province_code)
+            relevant_video_list.append(relevant_video_id)
     video_static_info = env_dict.get('vid_day_fea_list', [])
     video_hour_static_info = env_dict.get('vid_hour_fea_list', [])
     #print("video_static_info:",video_static_info)
@@ -184,6 +188,7 @@ def get_featurs(mid, data, size, top_K, flow_pool_P, env_dict=None):
     feature_dict["hour_ctr_score_list"] = hour_ctr_score_list
     feature_dict["city_code"] = city_list
     feature_dict["province_code"] = province_list
+    feature_dict["relevant_video_id"] = relevant_video_list
     return feature_dict
 
 
@@ -235,13 +240,14 @@ def get_tf_serving_sores(feature_dict):
                     #"hour_return_rate": feature_dict["hour_return_rate_list"],
                     #"hour_ctr_score": feature_dict["hour_ctr_score_list"],
                     "city_code": feature_dict['city_code'],
-                    "province_code": feature_dict['province_code']
+                    "province_code": feature_dict['province_code'],
+                    "relevant_video_id":feature_dict['relevant_video_id']
     }
     request_data_dict= {}
     request_data_dict["inputs"] = inputs_data
     #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))