From a324aa2c91bd88e1952b017ff3b489f919f3c3c3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Wed, 10 May 2023 22:39:12 +0200 Subject: [PATCH] Update 'train.py' --- train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index acf1871..ca3ab8e 100644 --- a/train.py +++ b/train.py @@ -6,6 +6,7 @@ from keras.applications.densenet import layers from sklearn.model_selection import train_test_split import tensorflow +EPOCHS = int(os.environ['EPOCHS']) train_data_x = pandas.read_csv('./X_train.csv') @@ -25,6 +26,6 @@ norm_games_model.compile( loss=tensorflow.keras.losses.MeanSquaredError(), optimizer=tensorflow.keras.optimizers.Adam()) -norm_games_model.fit(games_all, games_predict, epochs=500) +norm_games_model.fit(games_all, games_predict, epochs=EPOCHS) norm_games_model.save('test') \ No newline at end of file