update dllib-mlflow.py

2022-05-11 15:15:22 +02:00 · 2022-05-11 15:15:22 +02:00 · 73bd0a68c7
commit 73bd0a68c7
parent 851049fb38
1 changed files with 2 additions and 1 deletions
--- a/Biblioteka_DL/dllib-mlflow.py
+++ b/Biblioteka_DL/dllib-mlflow.py
@ -324,7 +324,8 @@ def my_main(epochs):

    # save model
    torch.save(model, "games_model.pkl")
-    return accuracy_score(labels_test_g, np.argmax(pred, axis=1))
+    acc = accuracy_score(labels_test_g, np.argmax(pred, axis=1))
+    return acc


 with mlflow.start_run() as run: