2024-06-18 15:53:06 +02:00
---
2025-06-20 19:46:38 +05:30
title: MLflow Connector | OpenMetadata ML Model Integration
2024-06-18 15:53:06 +02:00
slug: /connectors/ml-model/mlflow
---
{% connectorDetailsHeader
name="MLflow"
stage="PROD"
platform="OpenMetadata"
2025-02-03 18:58:13 +05:30
availableFeatures=["ML Features", "Hyperparameters", "ML Store"]
2024-06-18 15:53:06 +02:00
unavailableFeatures=[]
/ %}
In this section, we provide guides and references to use the MLflow connector.
Configure and schedule MLflow metadata and profiler workflows from the OpenMetadata UI:
- [Requirements ](#requirements )
- [Metadata Ingestion ](#metadata-ingestion )
2025-04-03 13:34:59 +05:30
- [Troubleshooting ](/connectors/ml-model/mlflow/troubleshooting )
2024-06-18 15:53:06 +02:00
2024-12-12 11:34:09 +05:30
{% partial file="/v1.7/connectors/ingestion-modes-tiles.md" variables={yamlPath: "/connectors/ml-model/mlflow/yaml"} /%}
2024-06-18 15:53:06 +02:00
## Requirements
To extract metadata, OpenMetadata needs two elements:
- **Tracking URI**: Address of local or remote tracking server. More information on the MLflow documentation [here ](https://www.mlflow.org/docs/latest/tracking.html#where-runs-are-recorded )
- **Registry URI**: Address of local or remote model registry server.
## Metadata Ingestion
{% partial
2024-12-12 11:34:09 +05:30
file="/v1.7/connectors/metadata-ingestion-ui.md"
2024-06-18 15:53:06 +02:00
variables={
connector: "Mlflow",
2024-12-12 11:34:09 +05:30
selectServicePath: "/images/v1.7/connectors/mlflow/select-service.png",
addNewServicePath: "/images/v1.7/connectors/mlflow/add-new-service.png",
serviceConnectionPath: "/images/v1.7/connectors/mlflow/service-connection.png",
2024-06-18 15:53:06 +02:00
}
/%}
{% stepsContainer %}
{% extraContent parentTagName="stepsContainer" %}
#### Connection Details
- **trackingUri**: Mlflow Experiment tracking URI. E.g., http://localhost:5000
- **registryUri**: Mlflow Model registry backend. E.g., mysql+pymysql://mlflow:password@localhost:3307/experiments
{% /extraContent %}
2024-12-12 11:34:09 +05:30
{% partial file="/v1.7/connectors/test-connection.md" /%}
2024-06-18 15:53:06 +02:00
2024-12-12 11:34:09 +05:30
{% partial file="/v1.7/connectors/ml-model/configure-ingestion.md" /%}
2024-06-18 15:53:06 +02:00
2024-12-12 11:34:09 +05:30
{% partial file="/v1.7/connectors/ingestion-schedule-and-deploy.md" /%}
2024-06-18 15:53:06 +02:00
{% /stepsContainer %}