From 95d3098bd3cc4aa51738d1cad40d62b58daa28f8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=83=AD=E7=BE=BD?= <guoyu@igengmei.com>
Date: Fri, 30 Jul 2021 23:41:06 +0800
Subject: [PATCH] =?UTF-8?q?service=20model=20=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
train/train_service.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train/train_service.py b/train/train_service.py
index 0cf6bf5..4c09926 100644
--- 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...")
--
2.18.0