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-11 22:40:39 +02:00
parent f2e53842b7
commit 691030d407
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -35,7 +35,7 @@ def my_main(EPOCHS, _run):
optimizer=tensorflow.keras.optimizers.Adam()) optimizer=tensorflow.keras.optimizers.Adam())
norm_games_model.fit(games_all, games_predict, epochs=EPOCHS) norm_games_model.fit(games_all, games_predict, epochs=EPOCHS)
accu = norm_games_modelevaluate(x_to_test, y_to_test) accu = norm_games_model.evaluate(x_to_test, y_to_test)
exint.add_artifact(filename=accu, name='metrics.txt') exint.add_artifact(filename=accu, name='metrics.txt')
norm_games_model.save('test') norm_games_model.save('test')