管道參數 - Amazon SageMaker

本文為英文版的機器翻譯版本，如內容有任何歧義或不一致之處，概以英文版為準。

管道參數

您可以使用參數將變數引入管道定義。您可以參考您在整個管道定義中定義的參數。參數具有預設值，您可以透過在啟動管道執行時指定參數值來覆寫預設值。預設值必須是與參數類型相符的執行個體。步驟定義中使用的所有參數都必須在管道定義中定義。本主題說明您可以定義的參數，以及如何實作這些參數。

Amazon SageMaker Pipelines 支援下列參數類型：

  • 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 SDK函數來處理參數，例如 sagemaker.workflow.functions.Join。如需參數的詳細資訊，請參閱 SageMaker 管道參數 。

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