Gerenciamento de modelos no MLflow Model Registry via Arena - Container Service for Kubernetes: Central de Documentação do Alibaba Cloud

Use a CLI do Arena para criar, consultar, atualizar e excluir modelos no MLflow Model Registry pela linha de comando.

Contexto

O MLflow é uma plataforma open source de gerenciamento do ciclo de vida de machine learning que permite rastrear treinamentos, gerenciar e implantar modelos. Para mais informações sobre o MLflow Model Registry, consulte MLflow Model Registry.

Pré-requisitos

Antes de começar, verifique se você tem:

Um cluster ACK Pro executando Kubernetes 1.20 ou posterior. Para mais informações, consulte Criar um cluster ACK Pro.
Um usuário do Resource Access Management (RAM) criado no console do RAM e associado a um grupo de cotas. Para mais informações, consulte Criar um usuário RAM e Etapa 1: Criar um grupo de cotas para o usuário RAM.
O componente MLflow implantado no namespace kube-ai do cluster ACK. Para mais informações, consulte Configure o MLflow Model Registry.
O cliente Arena configurado com a versão 0.9.14 ou posterior. Para mais informações, consulte Instale o cliente Arena.

Adicione uma versão de modelo

Execute o seguinte comando para adicionar uma versão de modelo:

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

Saída esperada:

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

Consulte modelos registrados ou versões de modelo

Consulte um modelo registrado

Execute o seguinte comando para consultar um modelo registrado chamado my-model:

arena model get \
    --name my-model

Saída esperada:

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

Consulte uma versão de modelo

Execute o seguinte comando para consultar a versão 1 do modelo my-model:

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

Saída esperada:

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

Liste todos os modelos registrados

Execute o seguinte comando para listar todos os modelos registrados:

arena model list

Atualize modelos registrados ou versões de modelo

Execute o seguinte comando para atualizar a descrição e as tags do modelo my-model:

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

Saída esperada:

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

Para excluir tags, adicione um hífen (-) ao final de cada chave ou par chave-valor da tag. O comando a seguir exclui key1 e key2 do modelo my-model, além de excluir key3 e key4 da versão 1:

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

Essa ação remove as tags key1 e key2 do modelo registrado my-model, bem como as tags key3 e key4 da versão 1.

Exclua modelos registrados ou versões de modelo

Aviso

A exclusão de um modelo registrado também remove todas as versões associadas a ele. Proceda com cautela.

Todos os comandos de exclusão solicitam confirmação antes da execução. Insira yes para confirmar ou no para cancelar.

Exclua um modelo registrado

Execute o seguinte comando para excluir o modelo my-model:

arena model delete \
    --name my-model

Para ignorar a solicitação de confirmação (por exemplo, em scripts automatizados), adicione a flag --force:

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

Exclua uma versão de modelo

Execute o seguinte comando para excluir a versão 1 do modelo my-model:

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

Para ignorar a solicitação de confirmação, adicione a flag --force:

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

Próximos passos

Para obter a referência completa de comandos de modelo do Arena, consulte o Guia de gerenciamento de modelos.