MLflow Integration#

autolog(*[, log_input_examples, ...])

Enable (or disable) and configure autologging from aimz to MLflow.

get_default_conda_env(*[, include_cloudpickle])

Return the default Conda environment for MLflow Models produced by this flavor.

get_default_pip_requirements(*[, ...])

Return the default pip requirements for MLflow Models produced by this flavor.

load_model(model_uri[, dst_path])

Load an aimz model from a local file or a run.

log_model(model[, artifact_path, conda_env, ...])

Log an aimz model as an MLflow artifact for the current run.

save_model(model, path[, conda_env, ...])

Save an aimz model to a path on the local file system.