Parámetros de canalizaciones - Amazon SageMaker

Parámetros de canalizaciones

Puede introducir variables en la definición de su canalización mediante parámetros. Puede hacer referencia a los parámetros que defina en la definición de la canalización. Los parámetros tienen un valor predeterminado, que puede anular si especifica valores para los parámetros al iniciar la ejecución de una canalización. El valor predeterminado debe ser una instancia que coincida con el tipo de parámetro. Todos los parámetros utilizados en las definiciones de los pasos deben estar definidos en la definición de la canalización. En este tema, se describen los parámetros que puede definir y cómo implementarlos.

Canalizaciones de Amazon SageMaker admite los siguientes tipos de parámetros:

  • ParameterString: representa un parámetro de cadena.

  • ParameterInteger: representa un parámetro entero.

  • ParameterFloat: representa un parámetro flotante.

  • ParameterBoolean: representa un tipo booleano de Python.

Los parámetros tienen el siguiente formato:

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

El siguiente ejemplo muestra una implementación de parámetro de muestra.

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

El parámetro se pasa al crear la canalización, tal como se muestra en el siguiente ejemplo.

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

También puede pasar un valor de parámetro diferente del valor predeterminado a una ejecución de canalización, como se muestra en el ejemplo siguiente.

execution = pipeline.start( parameters=dict( ProcessingInstanceCount="2", ModelApprovalStatus="Approved" ) )

Puede manipular los parámetros con funciones del SageMaker Python SDK como sagemaker.workflow.functions.Join. Para obtener más información sobre los parámetros, consulte SageMaker Pipelines Parameters.

Para conocer las limitaciones de los parámetros de Canalizaciones, consulte Limitations - Parameterization en el Amazon SageMaker Python SDK.