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

mlflow minor fix

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

4
avocado-mlflow.py
View File

@ -77,6 +77,6 @@ with mlflow.start_run():
mlflow.log_param("batch_size", batch_size) mlflow.log_param("batch_size", batch_size)
mlflow.log_metric("rmse", rmse) mlflow.log_metric("rmse", rmse)
#mlflow.keras.log_model(model, 'avocado_model.h5') #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.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(keras_model=model, path='avocado_model', 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])