From 761ce894fbf62812de318d67c757c463450231d3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Thu, 11 May 2023 21:30:50 +0200 Subject: [PATCH] Update 'train.py' --- train.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index fc96300..961924b 100644 --- a/train.py +++ b/train.py @@ -15,13 +15,12 @@ exint.observers.append(FileStorageObserver('my_runs')) def my_config(): EPOCHS = int(os.environ['EPOCHS']) train_data_x = exint.open_resource('./X_train.csv', 'r') - - normalize = layers.Normalization() _run.info["epochs"] = EPOCHS @exint.main -def my_main(EPOCHS,train_data_x, normalize): +def my_main(EPOCHS,train_data_x): + normalize = layers.Normalization() games_all = train_data_x.copy() games_predict = train_data_x.pop('User_Score') normalize.adapt(games_all)