管道参数 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异，请参阅中国的 Amazon Web Services 服务入门。

本文属于机器翻译版本。若本译文内容与英语原文存在差异，则一律以英文原文为准。

管道参数

您可以使用参数将变量引入管道定义中。您可以引用在整个管道定义中定义的参数。参数具有默认值，您可以通过在启动管道执行时指定参数值来覆盖该值。默认值必须是与参数类型匹配的实例。 步骤定义中使用的所有参数都必须在管道定义中定义。亚马逊 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(
        ProcessingInstanceType="ml.c5.xlarge",
        ModelApprovalStatus="Approved"
    )
)

你可以使用 SageMaker Python 开发工具包的功能 sagemaker.workflow.functions.Join. 有关参数的更多信息，请参阅 SageMaker 管道参数.