mlflow.sklearn

Sample MLflow integration for SciKit-Learn.

mlflow.sklearn.load_model(path, run_id=None)

Load a SciKit-Learn model from a local file (if run_id is None) or a run.

mlflow.sklearn.load_pyfunc(path)

mlflow.sklearn.log_model(sk_model, artifact_path)

Log a SciKit-Learn model as an MLflow artifact for the current run.

mlflow.sklearn.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.