美购精排模型

28556a05 · 郭羽 · 07096ada · 28556a05
Commit 28556a05 authored May 27, 2021 by 郭羽
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

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

    print("train start...")
    timestmp3 = int(round(time.time()))
-    # model = train(emb_columns,number_columns,oneHot_columns,inputs,train_data)
+    model = train(emb_columns,number_columns,oneHot_columns,inputs,train_data)
    timestmp4 = int(round(time.time()))
    print("train end...耗时h:{}".format((timestmp4 - timestmp3)/60/60))

-    evaluate(None,test_data)
+    evaluate(model,test_data)


    predict(model_file,test_data)