Skip to content

F
ffm-baseline
Commit 8d1dd5a7 authored by 张彦钊's avatar 张彦钊
Browse files
Options

change transform

parent 82ba92aa
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tensnsorflow/ffm.py
View file @ 8d1dd5a7
...@@ -265,13 +265,13 @@ def get_predict_set(ucity_id, cid,model): ...@@ -265,13 +265,13 @@ def get_predict_set(ucity_id, cid,model):
df = df.drop([0, "seq"], axis=1) df = df.drop([0, "seq"], axis=1)
print(df.head()) print(df.head())
native_pre = df[df["label"] == 0] native_pre = df[df["label"] == "0"]
native_pre = native_pre.drop("label", axis=1) native_pre = native_pre.drop("label", axis=1)
print("native_pre shape") print("native_pre shape")
print(native_pre.shape) print(native_pre.shape)
native_pre.to_csv(path + "native_pre.csv", index=None) native_pre.to_csv(path + "native_pre.csv", index=None)
nearby_pre = df[df["label"] == 1] nearby_pre = df[df["label"] == "1"]
nearby_pre = nearby_pre.drop("label", axis=1) nearby_pre = nearby_pre.drop("label", axis=1)
print("nearby_pre shape") print("nearby_pre shape")
print(nearby_pre.shape) print(nearby_pre.shape)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment