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