From 8df74249ffd704b6adb933dc3982743fd88d5e7e Mon Sep 17 00:00:00 2001 From: Agata Date: Sun, 22 May 2022 11:52:37 +0200 Subject: [PATCH] Update training script --- training_mlflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/training_mlflow.py b/training_mlflow.py index ea678e7..c783289 100644 --- a/training_mlflow.py +++ b/training_mlflow.py @@ -118,9 +118,9 @@ def my_main(epochs): tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme if tracking_url_type_store != "file": - mlflow.sklearn.log_model(model, "s444421", registered_model_name="s444421", signature=signature, input_example=X_test.numpy()[:5]) + mlflow.sklearn.log_model(model, "my_model", registered_model_name="s444421", signature=signature, input_example=X_test.numpy()[:5]) else: - mlflow.sklearn.log_model(model, "s444421", signature=signature, input_example=X_test.numpy()[:5]) + mlflow.sklearn.log_model(model, "my_model", signature=signature, input_example=X_test.numpy()[:5]) # In[ ]: