change transform

tensnsorflow/ffm.py

@@ -265,13 +265,13 @@ def get_predict_set(ucity_id, cid,model):
     df = df.drop([0, "seq"], axis=1)
     print(df.head())
 
-    native_pre = df[df["label"] == 0]
+    native_pre = df[df["label"] == "0"]
     native_pre = native_pre.drop("label", axis=1)
     print("native_pre shape")
     print(native_pre.shape)
     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)
     print("nearby_pre shape")
     print(nearby_pre.shape)