add na

6ab4a5be · 张彦钊 · 42b731eb · 6ab4a5be
Commit 6ab4a5be authored Jan 08, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

data2ffm.py eda/esmm/Feature_pipline/data2ffm.py +1 -1

No files found.
--- a/eda/esmm/Feature_pipline/data2ffm.py
+++ b/eda/esmm/Feature_pipline/data2ffm.py
@@ -224,7 +224,7 @@ def get_predict_set(model,columns):

    print(df.shape)
    for i in columns.keys():
-        df.loc[~df["city_id"].isin(columns[i]), [i]] = "na"
+        df.loc[~df[i].isin(columns[i]), [i]] = "na"
    print(df.shape)
    df["cid_id"] = df["cid_id"].astype("str")
    df["clevel1_id"] = df["clevel1_id"].astype("str")