Update 'train.py'

2023-05-11 22:40:39 +02:00 · 2023-05-11 22:40:39 +02:00 · 691030d407
commit 691030d407
parent f2e53842b7
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -35,7 +35,7 @@ def my_main(EPOCHS, _run):
        optimizer=tensorflow.keras.optimizers.Adam())

    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')
    norm_games_model.save('test')