From cda9bae6366d31b1e84cdaf415b967c21677ade6 Mon Sep 17 00:00:00 2001 From: patrycjalazna Date: Sun, 23 May 2021 23:55:22 +0200 Subject: [PATCH] mlflow minor fix --- avocado-mlflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/avocado-mlflow.py b/avocado-mlflow.py index 69cd613..1574019 100644 --- a/avocado-mlflow.py +++ b/avocado-mlflow.py @@ -77,6 +77,6 @@ with mlflow.start_run(): mlflow.log_param("batch_size", batch_size) mlflow.log_metric("rmse", rmse) #mlflow.keras.log_model(model, 'avocado_model.h5') - mlflow.keras.log_model(keras_model=model, artifact_path='avocado_model', registered_model_name="s434742", signature=infer_signature(X_train, y_train), input_example=X_train.iloc[0]) - mlflow.keras.save_model(keras_model=model, path='avocado_model', signature=infer_signature(X_train, y_train), input_example=X_train.iloc[0]) + mlflow.keras.log_model(model, 'avocado_model', registered_model_name="s434742", signature=infer_signature(X_train, y_train), input_example=X_train.iloc[0]) + mlflow.keras.save_model(model, 'avocado_model', signature=infer_signature(X_train, y_train), input_example=X_train.iloc[0])