delete time

tensnsorflow/ffm.py

@@ -156,12 +156,7 @@ def get_data():
     df = df.rename(columns={0: "y", 1: "z", 2: "stat_date", 3: "ucity_id",4: "clevel1_id", 5: "ccity_name",
                             6:"device_type",7:"manufacturer",8:"channel",9:"top",10:"time",11:"device_id"})
     print("esmm data ok")
-    # print(df.head(2))
-
-    liuxiao = df[(df["device_id"] == "358035085192742") & (df["y"] == 1)]
-    liuxiao.to_csv(path + "1beforeliuxiao.csv", index=False)
-    print(liuxiao.head())
-
+    # print(df.head(2)
 
     df["clevel1_id"] = df["clevel1_id"].astype("str")
     df["y"] = df["y"].astype("str")
@@ -169,8 +164,7 @@ def get_data():
     df["top"] = df["top"].astype("str")
     df["y"] = df["stat_date"].str.cat([df["device_id"].values.tolist(),df["y"].values.tolist(),df["z"].values.tolist()], sep=",")
 
-
-    df = df.drop(["z","stat_date","device_id"], axis=1).fillna(0.0)
+    df = df.drop(["z","stat_date","device_id","time"], axis=1).fillna("na")
     print(df.head(2))
     features = 0
     for i in ["ucity_id","clevel1_id","ccity_name","device_type","manufacturer","channel"]:
@@ -206,12 +200,8 @@ def transform(a,validate_date):
     test = test.drop("stat_date",axis=1)
     print("train shape")
     print(train.shape)
-    # train.to_csv(path + "tr.csv", sep="\t", index=False)
-    # test.to_csv(path + "va.csv", sep="\t", index=False)
-    liuxiao = df[(df["device_id"] == "358035085192742") & (df["y"]=="1")]
-    liuxiao.to_csv(path + "liuxiao.csv", index=False)
-    liuxiao = df[(df["device_id"] == "358035085192742") & (df["y"] == 1)]
-    liuxiao.to_csv(path + "liuxiao1.csv", index=False)
+    train.to_csv(path + "tr.csv", sep="\t", index=False)
+    test.to_csv(path + "va.csv", sep="\t", index=False)
 
     return model