MLflow integration for Keras.
Load a Keras model from a local file (if run_id is None) or a run.
Loads a Keras model as a PyFunc from the passed-in persisted Keras model file.
Parameters: model_file – Path to Keras model file. Returns: PyFunc model.
log_model(keras_model, artifact_path, **kwargs)
Log a Keras model as an MLflow artifact for the current run.
save_model(keras_model, path, conda_env=None, mlflow_model=<mlflow.models.Model object>)
Save a Keras model to a path on the local file system.
- keras_model – Keras model to be saved.
- path – Local path where the model is to be saved.
- mlflow_model – MLflow model config this flavor is being added to.