美购精排模型

mlp/train.py

@@ -47,17 +47,14 @@ def getDataVocabFromRedis(version):
 
 # 数据类型转换
 def csvTypeConvert(columns,df,data_vocab):
-    print(set(df["label"].tolist()))
-    # df["label"] = df["label"].astype("int")
+    df["label"] = df["label"].astype("int")
     for k in columns:
         # 离散na值填充
         if data_vocab.get(k):
-            if k == "item_first_solutions__1":
-                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].map(lambda x:x if is_float(x) else 0)
             df[k] = df[k].fillna(0)
             df[k] = df[k].astype("float")
     # print(df.dtypes)