From f2e53842b784c2ea059c02f95339e71220778b23 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Thu, 11 May 2023 22:39:41 +0200 Subject: [PATCH] Update 'train.py' --- train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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')