Update 'simple_regression_lab8.py'

2022-05-14 13:22:33 +02:00 · 2022-05-14 13:22:33 +02:00 · 1d74959d77
commit 1d74959d77
parent 0559fd6038
1 changed files with 1 additions and 1 deletions
--- 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))