美购精排模型

mlp/train.py

@@ -47,20 +47,18 @@ def getDataVocabFromRedis(version):
 
 # 数据类型转换
 def csvTypeConvert(columns,df,data_vocab):
+    print(set(df["label"].tolist()))
+    df["label"] = df["label"].astype("int")
     for k in columns:
         # 离散na值填充
         if data_vocab.get(k):
+            print(set(df[k].tolist()))
             df[k] = df[k].fillna("-1")
             df[k] = df[k].astype("string")
         elif k != "label":
             df[k] = df[k].map(lambda x:x if is_float(x) else 0)
             df[k] = df[k].fillna(0)
             df[k] = df[k].astype("float")
-
-    print(set(df["label"].tolist()))
-
-    df["label"] = df["label"].astype("int")
-
     # print(df.dtypes)
     return df