美购精排模型

a8ec78cc · 郭羽 · 8b7a6324 · a8ec78cc
Commit a8ec78cc authored May 29, 2021 by 郭羽
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

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

No files found.
--- a/mlp/train.py
+++ b/mlp/train.py
@@ -208,6 +208,7 @@ if __name__ == '__main__':

    # 获取训练列
    columns = df_train.columns.tolist()
+    print(columns)
    emb_columns,number_columns,oneHot_columns, datasColumns,inputs = getTrainColumns(columns, data_vocab)

    df_train = df_train[datasColumns + ["label"]]