Python API

MLflow’s Python API is organized into the following packages. The most common functions are also exposed in the mlflow module, so we recommend starting there.

• mlflow
• mlflow.azureml
• mlflow.entities
• mlflow.h2o
• mlflow.keras
• mlflow.projects
• mlflow.pyfunc
• mlflow.pytorch
• mlflow.sagemaker
• mlflow.models
• mlflow.sklearn
• mlflow.spark
• mlflow.tensorflow
• mlflow.tracking

See also an index of all functions and classes.