MLflow AI Model asset

The MLflow integration of Collibra Platform Self-Hosted uses 1 asset type. The asset type comes out of the box with your software.

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
MLflow AI Model Version

A subtype of AI Model Version that represents AI model versions in MLflow.

MLflow diagram view

For information on the data that is integrated from MLflow, go to Integrated MLflow data.

Steps

  1. Open the asset page.
  2. Click the Diagram tab.
    The diagram is shown in the default diagram view.
  3. Click to add a new view.
  4. Select the Text option below the diagram view name.
    The diagram view text editor is shown.
  5. Copy the code from the Show JSON code section below and paste it in the diagram view text editor.
  6. Click Save.
  7. Edit the name and description of the diagram view as needed.