From e6bc92507c542c2b1f86ecfa4b4505edc772b5dc Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 01:38:39 +0800 Subject: [PATCH] update --- .idea/workspace.xml | 36 +++++++++++++++++++----------------- serve.py | 3 +-- 2 files changed, 20 insertions(+), 19 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index b301a65..b7115b5 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -1,7 +1,9 @@ - + + + @@ -207,7 +210,7 @@ - + @@ -248,16 +251,15 @@ - - @@ -358,8 +360,8 @@ - - + + @@ -368,11 +370,11 @@ - - + + - + diff --git a/serve.py b/serve.py index 1dfd2a9..b56a3ab 100644 --- a/serve.py +++ b/serve.py @@ -177,13 +177,12 @@ def train_models(): train_X, valid_X, train_Y, valid_Y = data_gainer.get_train_and_valid_data() - print(train_X, valid_X, train_Y, valid_Y) - print(train_X.shape[0]) if train_X.shape[0] < 500: config.batch_size = 32 if train_X.shape[0] < 200: config.batch_size = 16 + print(train_X[:, :1], valid_X[:, :1], train_Y, valid_Y) train(config, train_X[:, :1], train_Y, valid_X[:, :1], valid_Y) lock.release()