MLflow integration for scikit-learn.
Load a scikit-learn model from a local file (if
run_idis None) or a run.
Log a scikit-learn model as an MLflow artifact for the current run.
save_model(sk_model, path, conda_env=None, mlflow_model=<mlflow.models.Model object>)
Save a scikit-learn model to a path on the local file system.
- sk_model – scikit-learn model to be saved.
- path – Local path where the model is to be saved.
- conda_env – Path to a Conda environment file. If provided, this decribes the environment this model should be run in. At minimum, it should specify python, scikit-learn, and mlflow with appropriate versions.
- mlflow_model – MLflow model config this flavor is being added to.