service model 优化

95d3098b · 郭羽 · 74eceee2 · 95d3098b
Commit 95d3098b authored Jul 30, 2021 by 郭羽
Show 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
@@ -112,7 +112,7 @@ def train(emb_columns, number_columns, inputs, train_dataset):

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