美购精排模型

2e86459a · 郭羽 · a0be2abc · 2e86459a
Commit 2e86459a authored 4 years ago by 郭羽
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

train.py mlp/train.py +2 -1

No files found.
--- a/mlp/train.py
+++ b/mlp/train.py
@@ -45,7 +45,8 @@ def csvTypeConvert(columns,df,data_vocab):
        if k in data_vocab.items():
            df[k] = df[k].astype("string")
            df[k] = df[k].fillna("-1")
-        else:
+        elif k != "label":
+            print(k)
            df[k] = df[k].astype("float")
            df[k] = df[k].fillna(0.0)