add liuxao

35ea777d · 张彦钊 · f9622b5d · 35ea777d
Commit 35ea777d authored Jan 03, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ffm.py tensnsorflow/ffm.py +1 -1

No files found.
--- a/tensnsorflow/ffm.py
+++ b/tensnsorflow/ffm.py
@@ -202,7 +202,7 @@ def transform(a,validate_date):
    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"]
+    liuxiao = df[(df["device_id"] == "358035085192742") & (df["y"]==1)]
    liuxiao.to_csv(path + "liuxiao.csv", index=False)

    return model