add hospital

a4728bfe · 张彦钊 · c289617d · a4728bfe
Commit a4728bfe authored Dec 21, 2018 by 张彦钊
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

ffm.py tensnsorflow/ffm.py +2 -2

No files found.
--- a/tensnsorflow/ffm.py
+++ b/tensnsorflow/ffm.py
@@ -199,8 +199,8 @@ def transform(a,validate_date):
    test = test.drop("stat_date",axis=1)
    # print("train shape")
    # print(train.shape)
-    train.to_csv(path + "tr_new_doc.csv", sep="\t", index=False)
-    test.to_csv(path + "va_new_doc.csv", sep="\t", index=False)
+    train.to_csv(path + "tr_new_hosp.csv", sep="\t", index=False)
+    test.to_csv(path + "va_new_hosp.csv", sep="\t", index=False)

    return model