From 26ddaa3a0c43cd3da384e1976107cb8241e01ae8 Mon Sep 17 00:00:00 2001 From: s434695 Date: Sun, 16 May 2021 21:50:54 +0200 Subject: [PATCH] fix train --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 9598ce2..6d2ae90 100755 --- a/train.py +++ b/train.py @@ -26,7 +26,7 @@ def my_config(): batch_param = int(sys.argv[1]) epoch_param = int(sys.argv[2]) -def regression_model(epoch_param, batch_param, _run): +def prepare_model(epoch_param, batch_param, _run): _run.info["prepare_model_ts"] = str(datetime.now()) # odczytanie danych z plików vgsales_train = pd.read_csv('train.csv') @@ -66,7 +66,7 @@ def regression_model(epoch_param, batch_param, _run): @ex.main def my_main(epoch_param, batch_param): - print(regression_model()) + print(prepare_model()) r = ex.run() ex.add_artifact("vgsales_model.h5") \ No newline at end of file