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

input example corrected
All checks were successful
s434742-training/pipeline/head This commit looks good
Details
s434742-evaluation/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
patrycjalazna 2021-05-24 08:40:36 +02:00
parent a0d9a89c69
commit 88cf23958f
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(model, '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=np.array(X_train.iloc[0]))
mlflow.keras.save_model(model, '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=np.array(X_train.iloc[0]))