s434686/ium_z434686
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'train.py'

Browse Source
This commit is contained in:
Patryk Gałka 2023-05-10 22:39:12 +02:00
parent d07853cce0
commit a324aa2c91
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -6,6 +6,7 @@ from keras.applications.densenet import layers
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split
import tensorflow import tensorflow
EPOCHS = int(os.environ['EPOCHS'])
train_data_x = pandas.read_csv('./X_train.csv') train_data_x = pandas.read_csv('./X_train.csv')
@ -25,6 +26,6 @@ norm_games_model.compile(
loss=tensorflow.keras.losses.MeanSquaredError(), loss=tensorflow.keras.losses.MeanSquaredError(),
optimizer=tensorflow.keras.optimizers.Adam()) 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') norm_games_model.save('test')