修改文件保存路径

tensnsorflow/ffm.py

@@ -68,8 +68,10 @@ def transform(df):
     print("size")
     print(df.shape)
     print(df.head())
-    train = df[df["stat_date"] != "2018-11-25"].drop("stat_date",axis=1)
-    test = df[df["stat_date"] == "2018-11-25"].drop("stat_date",axis=1)
+    train = df[df["stat_date"] != "2018-11-25"]
+    train = train.drop("stat_date",axis=1)
+    test = df[df["stat_date"] == "2018-11-25"]
+    test = test.drop("stat_date",axis=1)
     train.to_csv(path+"train.csv",index=None)
     test.to_csv(path + "test.csv", index=None)
     # yconnect = create_engine('mysql+pymysql://root:3SYz54LS9#^9sBvC@10.66.157.22:4000/jerry_test?charset=utf8')