排序模型优化

bfaec84e · 郭羽 · bfa8847f · bfaec84e
Commit bfaec84e authored Nov 10, 2021 by 郭羽
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

train_service.py train/train_service.py +1 -1

No files found.
--- a/train/train_service.py
+++ b/train/train_service.py
@@ -115,7 +115,7 @@ def train(emb_columns, number_columns, inputs, train_dataset):

    # train the model
    print("train start...")
-    model.fit(train_dataset, epochs=5)
+    model.fit(train_dataset, epochs=10)
    print("train end...")
    print("train save...")