diff --git a/train.py b/train.py index 46f6991..cfc3e9c 100644 --- 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')