The MLflow AI integration uses the following asset types. The asset types come out of the box with Collibra Platform.
Important The functions of MLflow integration is limited if you do not enable AI Governance. When you enable AI Governance, you can add custom attributes or custom relations, for example, AI Model Version, to the MLflow AI Model Version asset.
| Asset type | Description |
|---|---|
| AI Base Model |
The foundational business concept that defines the model's purpose and governing principles, remaining constant across all technical iterations. |
| MLflow AI Model Version |
A subtype of AI Model Version that represents AI model versions in MLflow. Only model versions with a |
| AI Agent | An autonomous system that perceives its environment, makes decisions based on predefined rules or learned patterns, and takes actions to achieve specific goals. |
| AI Agent Version | A specific, immutable implementation of the model logic and parameters that serves as the audited "raw" model version before it is activated. |