MLflow Integration#

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

Enable and configure autologging for aimz with MLflow.

get_default_conda_env(*[, include_cloudpickle])

Return the default Conda environment for saving an aimz model.

get_default_pip_requirements(*[, ...])

Return the default pip requirements for saving an aimz model.

load_model(model_uri[, dst_path])

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

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

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.