From 84641091ce4dd643d4f0f014a50d3160dd28a321 Mon Sep 17 00:00:00 2001 From: s434695 Date: Mon, 24 May 2021 00:42:36 +0200 Subject: [PATCH] fix --- vgsales-mlflow.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vgsales-mlflow.py b/vgsales-mlflow.py index 713b6bd..399b42e 100644 --- a/vgsales-mlflow.py +++ b/vgsales-mlflow.py @@ -70,3 +70,4 @@ with mlflow.start_run(): mlflow.log_metric("rmse", rmse) #mlflow.keras.log_model(model, 'vgsales_model.h5') mlflow.keras.save_model(model, "my_model", signature=mlflow.models.signature.infer_signature(x_train, y_train), input_example=x_train) + mlflow.keras.log_model(model, "model",registered_model_name="s434695", signature= mlflow.models.signature.infer_signature(x_train, y_train), input_example= x_train[0]) \ No newline at end of file