linfan 1 年間 前
コミット
fda8445fc5
1 ファイル変更8 行追加4 行削除
  1. 8 4
      rank_service.py

+ 8 - 4
rank_service.py

@@ -30,8 +30,12 @@ def get_featurs(mid, data, size, top_K, flow_pool_P, env_dict=None):
         recommendsource = env_dict.get('recommendsource', '0')
         machineinfo_platform = env_dict.get('machineinfo_platform', '')
         sencetype = env_dict.get('sencetype', '')
-        city_code = env_dict.get('city_code', -1)
-        province_code = env_dict.get('province_code', -1)
+        try:
+            city_code = int(env_dict.get('city_code', -1))
+            province_code = int(env_dict.get('province_code', -1))
+        except:
+            city_code = -1
+            province_code = -1
     mid_list = []
     videoid_list = []
     apptype_list = []
@@ -69,8 +73,8 @@ def get_featurs(mid, data, size, top_K, flow_pool_P, env_dict=None):
             recommendsource_list.append(recommendsource)
             machineinfo_platform_list.append(machineinfo_platform)
             sencetype_list.append(sencetype)
-            city_list.append(int(city_code))
-            province_list.append(int(province_code))
+            city_list.append(city_code)
+            province_list.append(province_code)
     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)