Construire un SageMaker XGBoost estimateur à l'aide de la règle XGBoost Debugger Report - Amazon SageMaker

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Construire un SageMaker XGBoost estimateur à l'aide de la règle XGBoost Debugger Report

La règle CreateXgboostReport collecte les tenseurs de sortie suivants à partir de votre tâche d'entraînement :

  • hyperparameters : enregistre à la première étape.

  • metrics : enregistre la perte et la précision toutes les 5 étapes.

  • feature_importance : enregistre toutes les 5 étapes.

  • predictions : enregistre toutes les 5 étapes.

  • labels : enregistre toutes les 5 étapes.

Les tenseurs de sortie sont enregistrés dans un compartiment S3 par défaut. Par exemple, s3://sagemaker-<region>-<12digit_account_id>/<base-job-name>/debug-output/.

Lorsque vous créez un SageMaker estimateur pour un poste de XGBoost formation, spécifiez la règle comme indiqué dans l'exemple de code suivant.

Using the SageMaker generic estimator
import boto3
import sagemaker
from sagemaker.estimator import Estimator
from sagemaker import image_uris
from sagemaker.debugger import Rule, rule_configs

rules=[
    Rule.sagemaker(rule_configs.create_xgboost_report())
]

region = boto3.Session().region_name
xgboost_container=sagemaker.image_uris.retrieve("xgboost", region, "1.2-1")

estimator=Estimator(
    role=sagemaker.get_execution_role()
    image_uri=xgboost_container,
    base_job_name="debugger-xgboost-report-demo",
    instance_count=1,
    instance_type="ml.m5.2xlarge",
    
    # Add the Debugger XGBoost report rule
    rules=rules
)

estimator.fit(wait=False)