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:39:41 +02:00
parent 29219cd795
commit f2e53842b7
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

@ -35,7 +35,8 @@ 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)
exint.add_artifact(filename=accu, name='metrics.txt')
norm_games_model.save('test') norm_games_model.save('test')