From a1300fe809830a7e6b7916fdc5edd6bcd96bd30c Mon Sep 17 00:00:00 2001 From: michalzareba Date: Sun, 23 May 2021 17:15:38 +0200 Subject: [PATCH] Add model registry --- lab08_mfl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lab08_mfl.py b/lab08_mfl.py index b438d49..a49e95d 100644 --- a/lab08_mfl.py +++ b/lab08_mfl.py @@ -82,7 +82,7 @@ with mlflow.start_run(): mlflow.log_metric("RMSE", rmse) signature = mlflow.models.signature.infer_signature(X_train.values, model.predict(X_train.values)) - mlflow.keras.save_model(model, "movies_imdb", input_example=input_example, signature=signature) + # mlflow.keras.save_model(model, "movies_imdb", input_example=input_example, signature=signature) tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme if tracking_url_type_store != "file":