管道參數

您可以使用參數將變數引入管道定義。您可以參考您在整個管道定義中定義的參數。參數具有預設值，您可以透過在啟動管道執行時指定參數值來覆寫預設值。預設值必須是與參數類型相符的執行個體。步驟定義中使用的所有參數都必須在管道定義中定義。Amazon SageMaker 模型建置管道支援下列參數類型：

• ParameterString – 表示字串參數。

• ParameterInteger – 表示整數參數。

• ParameterFloat – 表示浮點數參數。

• ParameterBoolean – 表示布林值 Python 類型。

參數的格式如下：

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

以下範例顯示範例參數實作。

from sagemaker.workflow.parameters import (
    ParameterInteger,
    ParameterString,
    ParameterFloat,
    ParameterBoolean
)

processing_instance_count = ParameterInteger(
    name="ProcessingInstanceCount",
    default_value=1
)

您可以在建立管道時傳遞參數，如下列範例所示。

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

您也可以將不同於預設值的參數值傳遞給管道執行，如下列範例所示。

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

您可以 SageMaker Python 用像 sagemaker.workflow.functions.Join. 如需有關參數的詳細資訊，請參閱 SageMaker 配管參數。

如需 SageMaker 管道參數的已知限制，請參閱 Amazon SageMaker Python 開發套件中的限制-參數化。