Model management in MLflow Model Registry through Arena - Container Service for Kubernetes

Utilisez l'interface de ligne de commande (CLI) Arena pour créer, interroger, mettre à jour et supprimer des modèles dans MLflow Model Registry directement depuis le terminal.

Contexte

MLflow est une plateforme open source de gestion du cycle de vie de l'apprentissage automatique, conçue pour suivre l'entraînement des modèles, gérer les modèles et les déployer. Pour plus d'informations sur MLflow Model Registry, consultez la documentation MLflow Model Registry.

Prérequis

Avant de commencer, assurez-vous de disposer des éléments suivants :

Un cluster ACK Pro exécutant Kubernetes 1.20 ou une version ultérieure. Pour plus d'informations, consultez la rubrique Créer un cluster ACK Pro.
Un utilisateur RAM créé dans la console RAM et associé à un groupe de quotas. Pour plus d'informations, consultez les rubriques Créer un utilisateur RAM et Étape 1 : Créer un groupe de quotas pour l'utilisateur RAM.
Le composant MLflow déployé dans le namespace kube-ai du cluster ACK. Pour plus d'informations, consultez la documentation Configurer MLflow Model Registry.
Le client Arena configuré avec la version 0.9.14 ou ultérieure. Pour plus d'informations, consultez la documentation Configurer le client Arena.

Ajouter une version de modèle

Exécutez la commande suivante pour ajouter une version de modèle :

arena model create \
    --name my-model \
    --tags key1,key2=value2 \
    --description "This is some description about my-model" \
    --version-tags key3,key4=value4 \
    --version-description "This is some description about my-model v1" \
    --source pvc://my-pvc/models/my-model/1

Résultat attendu :

INFO[0000] registered model "my-model" created
INFO[0000] model version 1 for "my-model" created

Interroger les modèles enregistrés ou les versions de modèle

Interroger un modèle enregistré

Exécutez la commande suivante pour interroger le modèle enregistré nommé my-model :

arena model get \
    --name my-model

Résultat attendu :

Name:                my-model
LatestVersion        1
CreationTime:        2024-04-29T16:15:26+08:00
LastUpdatedTime:     2024-04-29T16:44:17+08:00
Description:
  This is some description about my-model!
Tags:
  key1:
  key2: value2
Versions:
  Version    Source
  ---        ---
  1          pvc://my-pvc/models/my-model/1

Interroger une version de modèle

Exécutez la commande suivante pour interroger la version 1 du modèle my-model :

arena model get \
    --name my-model \
    --version 1

Résultat attendu :

Name:                my-model
Version:             1
CreationTime:        2024-04-29T16:42:18+08:00
LastUpdatedTime:     2024-04-29T16:42:18+08:00
Source:              pvc://my-pvc/models/my-model/1
Description:
  This is some description about my-model v1
Tags:
  createdBy: arena
  key3:
  key4: value4

Lister tous les modèles enregistrés

Exécutez la commande suivante pour lister tous les modèles enregistrés :

arena model list

Mettre à jour les modèles enregistrés ou les versions de modèle

Exécutez la commande suivante pour mettre à jour la description et les tags du modèle my-model :

arena model update \
    --name my-model \
    --description "This is some updated description" \
    --tags key1=updatedValue1,key2=updatedValue2

Résultat attendu :

INFO[0000] model version "my-model/1" updated

Pour supprimer des tags, ajoutez un tiret (-) à chaque clé de tag ou paire clé-valeur. La commande suivante supprime les tags key1 et key2 du modèle my-model, ainsi que les tags key3 et key4 de la version 1 :

arena model update \
    --name my-model \
    --tags key1-,key2=value2- \
    --version 1 \
    --version-tags key3-,key4=value4-

Cette opération retire les tags key1 et key2 du modèle enregistré my-model, et supprime les tags key3 et key4 de la version 1.

Supprimer les modèles enregistrés ou les versions de modèle

Avertissement

La suppression d'un modèle enregistré entraîne également la suppression de toutes les versions de modèle associées. Procédez avec prudence.

Toutes les commandes de suppression demandent une confirmation avant l'exécution. Saisissez yes pour confirmer ou no pour annuler.

Supprimer un modèle enregistré

Exécutez la commande suivante pour supprimer le modèle my-model :

arena model delete \
    --name my-model

Pour ignorer la demande de confirmation, par exemple dans des scripts automatisés, ajoutez l'option --force :

arena model delete \
    --name my-model \
    --force

Supprimer une version de modèle

Exécutez la commande suivante pour supprimer la version 1 du modèle my-model :

arena model delete \
    --name my-model \
    --version 1

Pour ignorer la demande de confirmation, ajoutez l'option --force :

arena model delete \
    --name my-model \
    --version 1 \
    --force

Étapes suivantes

Pour consulter la référence complète des commandes de modèle Arena, reportez-vous au Guide de gestion des modèles.