Paramètres du pipeline - Amazon SageMaker

Paramètres du pipeline

Vous pouvez introduire des variables dans la définition de votre pipeline à l'aide de paramètres. Les paramètres que vous définissez peuvent être référencés tout au long de votre définition de pipeline. Les paramètres ont une valeur par défaut, que vous pouvez remplacer en spécifiant des valeurs de paramètre lors du démarrage d'une exécution de pipeline. La valeur par défaut doit être une instance correspondant au type de paramètre. Tous les paramètres utilisés dans les définitions d'étape doivent être définis dans votre définition de pipeline. Amazon SageMaker Model Building Pipelines prend en charge les types de paramètre suivants :

  • ParameterString – Représentation d'un type str Python.

  • ParameterInteger – Représentation d'un type int Python.

  • ParameterFloat – Représentation d'un type float Python.

Les paramètres prennent le format suivant :

<parameter> = <parameter_type>( name="<parameter_name>", default_value=<default_value> )

Voici un exemple de mise en œuvre de paramètre :

from sagemaker.workflow.parameters import ( ParameterInteger, ParameterString, ParameterFloat ) processing_instance_count = ParameterInteger( name="ProcessingInstanceCount", default_value=1 )

Vous transmettez le paramètre lors de la création de votre pipeline comme suit :

pipeline = Pipeline( name=pipeline_name, parameters=[ processing_instance_count ], steps=[step_process] )

Vous pouvez également transmettre une valeur de paramètre qui diffère de la valeur par défaut à une exécution de pipeline comme suit :

execution = pipeline.start( parameters=dict( ProcessingInstanceType="ml.c5.xlarge", ModelApprovalStatus="Approved" ) )