minor fix
This commit is contained in:
parent
00fcfd4d1c
commit
ece90d90e9
@ -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])
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user