Etapy modelu

Wprowadzenie do MLflow

Weston Bassler

Senior MLOps Engineer

Środowiska programistyczne

Diagram środowisk: development, staging i production

Wprowadzenie do MLflow

Etapy modelu w MLflow

  • Przypisywane do wersji modelu

  • Predefiniowane etapy:

    • None
    • Staging
    • Production
    • Archived

  • Jeden etap w danym momencie

Wprowadzenie do MLflow

Predefiniowane etapy

Predefiniowane etapy

Wprowadzenie do MLflow

None

Etap None

Wprowadzenie do MLflow

Staging

Etap Staging

Wprowadzenie do MLflow

Production

Etap produkcyjny

Wprowadzenie do MLflow

Archived

Etap archiwalny

Wprowadzenie do MLflow

Przejście modelu między etapami

transition-ui

# Import MLFlow Client
from mlflow import MlflowClient
client = MlflowClient()


# Transition to Staging
client.transition_model_version_stage(
        name="Unicorn",
        version=3,
        stage="Staging"
    )
Wprowadzenie do MLflow

Przejście wersji modelu do Staging

# Transition to Staging
client.transition_model_version_stage(name="Unicorn", version=3, stage="Staging")

<ModelVersion: creation_timestamp=1679693470034, current_stage='Staging', 
description=None, last_updated_timestamp=1679699050734, name='Unicorn', 
run_id='a1454f2865e449f8835f38f71e53e547', run_link=None, 
source='./mlruns/1/a1454f2865e449f8835f38f71e53e547/artifacts/model', 
status='READY', status_message=None, tags={}, user_id=None, version=3>
Wprowadzenie do MLflow

Interfejs rejestru

staging-ui

Wprowadzenie do MLflow

Przejście do Production

# Transition to Production
client.transition_model_version_stage(name="Unicorn", version=3, 
                                      stage="Production")

<ModelVersion: creation_timestamp=1679693470034, current_stage='Production', 
description=None, last_updated_timestamp=1679699633297, name='Unicorn', 
run_id='a1454f2865e449f8835f38f71e53e547', run_link=None, 
source='./mlruns/1/a1454f2865e449f8835f38f71e53e547/artifacts/model', 
status='READY', status_message=None, tags={}, user_id=None, version=3>
Wprowadzenie do MLflow

Czas na ćwiczenia!

Wprowadzenie do MLflow

Preparing Video For Download...