Các giai đoạn của mô hình

Giới thiệu về MLflow

Weston Bassler

Senior MLOps Engineer

Môi trường phần mềm

Sơ đồ môi trường development, staging và production

Các giai đoạn mô hình MLflow

Gán cho các phiên bản mô hình
Giai đoạn dựng sẵn:
- None
- Staging
- Production
- Archived
Chỉ một giai đoạn tại một thời điểm

Giai đoạn dựng sẵn

None

Giai đoạn None

Staging

Giai đoạn Staging

Production

Giai đoạn Production

Archived

Giai đoạn Archived

Chuyển giai đoạn mô hình

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"
    )

Chuyển phiên bản sang 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>

Giao diện Registry

staging-ui

Chuyển sang 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>

Ayo berlatih!

Giới thiệu về MLflow