From e10b25c9c95eafc41c362a12e36c91bbc8acf01a Mon Sep 17 00:00:00 2001 From: Maciej Czajka Date: Wed, 11 May 2022 15:17:17 +0200 Subject: [PATCH] update dllib-mlflow.py --- Biblioteka_DL/dllib-mlflow.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/Biblioteka_DL/dllib-mlflow.py b/Biblioteka_DL/dllib-mlflow.py index 56543e8..36c3bde 100644 --- a/Biblioteka_DL/dllib-mlflow.py +++ b/Biblioteka_DL/dllib-mlflow.py @@ -316,7 +316,7 @@ def my_main(epochs): pred = pred.detach().numpy() print("The accuracy is", accuracy_score(labels_test_g, np.argmax(pred, axis=1))) - # mlflow.log_metric("accuracy", accuracy_score(labels_test_g, np.argmax(pred, axis=1))) + mlflow.log_metric("accuracy", accuracy_score(labels_test_g, np.argmax(pred, axis=1))) pred = pd.DataFrame(pred) @@ -324,11 +324,9 @@ def my_main(epochs): # save model torch.save(model, "games_model.pkl") - acc = accuracy_score(labels_test_g, np.argmax(pred, axis=1)) - return acc + with mlflow.start_run() as run: - acc = my_main(epochs) - mlflow.log_param("epochs", epochs) - mlflow.log_metric("accuracy", acc) \ No newline at end of file + my_main(epochs) + mlflow.log_param("epochs", epochs) \ No newline at end of file