s434742/ium_434742
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

minor fix

Browse Source
This commit is contained in:
patrycjalazna 2021-05-23 23:25:07 +02:00
parent 00fcfd4d1c
commit ece90d90e9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
avocado-mlflow.py
View File

@ -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, path='avocado_model', registered_model_name="s434742", signature=infer_signature(X_train, y_train), input_example=X_train.iloc[0])
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])