Parámetros de canalización - Amazon SageMaker

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Parámetros de canalización

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 se pueden definir y cómo implementarlos.

Amazon SageMaker Pipelines 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 parámetros con SDK funciones de SageMaker Python como sagemaker.workflow.functions.Join. Para obtener más información sobre los parámetros, consulte Parámetros de SageMaker canalizaciones.

Para conocer las limitaciones conocidas de los parámetros de las canalizaciones, consulte Limitaciones: parametrización en Amazon Python. SageMaker SDK