Python API

The MLflow Python API is organized into the following modules. 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.mleap
• mlflow.models
• mlflow.projects
• mlflow.pyfunc
  • Filesystem format
    • MLModel configuration
  • Inference API
  • Creating custom Pyfunc models
    • Workflows
    • Which workflow is right for my use case?
• mlflow.pytorch
• mlflow.sagemaker
• mlflow.sklearn
• mlflow.spark
• mlflow.tensorflow
• mlflow.tracking

See also an index of all functions and classes.