hariishaa 622816dcb8
feat(metadata-ingestion): implement mlflow source (#7971)
Co-authored-by: Andrew Sikowitz <andrew.sikowitz@acryl.io>
2023-09-26 13:51:30 -04:00

1.9 KiB

Concept Mapping

This ingestion source maps the following MLflow Concepts to DataHub Concepts:

Source Concept DataHub Concept Notes
Registered Model MlModelGroup The name of a Model Group is the same as a Registered Model's name (e.g. my_mlflow_model)
Model Version MlModel The name of a Model is {registered_model_name}{model_name_separator}{model_version} (e.g. my_mlflow_model_1 for Registered Model named my_mlflow_model and Version 1, my_mlflow_model_2, etc.)
Model Stage Tag The mapping between Model Stages and generated Tags is the following:
- Production: mlflow_production
- Staging: mlflow_staging
- Archived: mlflow_archived
- None: mlflow_none