ssachkovskaya / example-gto-artifactsyaml

Get Started GTO Project

Example GTO Model Registry

A model registry is a tool to catalog ML models and their versions. Models from your data science projects can be discovered, tested, shared, deployed, and audited from there. DVC, GTO, and MLEM enable these capabilities on top of Git, so you can stick to an existing software engineering stack.

This repo is an example of Model Registry built with these tools. The model dashboard:

$ gto show
╒══════════╤══════════╤═════════╤═════════╤════════════╕
│ name     │ latest   │ #dev    │ #prod   │ #staging   │
╞══════════╪══════════╪═════════╪═════════╪════════════╡
│ churn    │ v3.1.1   │ v3.1.1  │ v3.0.0  │ v3.1.0     │
│ segment  │ v0.4.1   │ v0.4.1  │ -       │ -          │
│ cv-class │ v0.1.13  │ -       │ -       │ -          │
╘══════════╧══════════╧═════════╧═════════╧════════════╛

• The latest column shows the latest model versions,
• The #dev column represent model versions promoted to a Stage dev (same for #prod and #staging),
• Versions are registered and promoted to Stages by Git tags - you can click the links to see the which specific Git tag did it,
• Artifact metadata like path and description is stored in artifacts.yaml,
• Github Actions page of this repo have examples of workflows where we act upon these Git tags.

Branch mlem contains a version that also uses MLEM to deploy a model in CI/CD upon deployment stage assignment. Check out the deployed service at http://mlem-dev.herokuapp.com/docs.

🧑‍💻 To continue learning, head to Get Started with GTO.