修改数据写入

tensnsorflow/feature_engineering.py

@@ -119,6 +119,9 @@ def get_predict(date,value_map):
     print("predict shape")
     print(df.shape)
 
+    df["uid"] = df["device_id"]
+    df["city"] = df["ucity_id"]
+
     features = ["ucity_id", "clevel1_id", "ccity_name", "device_type", "manufacturer",
                 "channel", "top", "time", "stat_date","device_id"]
     for i in features:
@@ -135,10 +138,7 @@ def get_predict(date,value_map):
     native_pre = native_pre.drop("label", axis=1)
     nearby_pre = df[df["label"] == 1]
     nearby_pre = nearby_pre.drop("label", axis=1)
-    native_pre["uid"] = native_pre["device_id"]
-    native_pre["city"] = native_pre["ucity_id"]
-    nearby_pre["uid"] = nearby_pre["device_id"]
-    nearby_pre["city"] = nearby_pre["ucity_id"]
+
 
     for i in ["ucity_id", "clevel1_id", "ccity_name", "device_type", "manufacturer",
                 "channel", "top", "l1", "time", "stat_date","l2","device_id"]: