Update 'train.py'

2023-05-11 22:39:41 +02:00 · 2023-05-11 22:39:41 +02:00 · f2e53842b7
commit f2e53842b7
parent 29219cd795
1 changed files with 2 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -35,7 +35,8 @@ 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)
+    exint.add_artifact(filename=accu, name='metrics.txt')
    norm_games_model.save('test')