From 1d74959d774198ec3312b36a91721c9cf6954312 Mon Sep 17 00:00:00 2001 From: Kacper Dudzic Date: Sat, 14 May 2022 13:22:33 +0200 Subject: [PATCH] Update 'simple_regression_lab8.py' --- simple_regression_lab8.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/simple_regression_lab8.py b/simple_regression_lab8.py index 43573b5..a5f1e00 100644 --- a/simple_regression_lab8.py +++ b/simple_regression_lab8.py @@ -89,7 +89,7 @@ def train(): tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme if tracking_url_type_store != 'file': mlflow.keras.log_model(model, 'lego-model', registered_model_name='TFLegoModel', - signature=signature) + signature=signature, input_example=np.array(500)) else: mlflow.keras.log_model(model, 'model', signature=signature, input_example=np.array(500))