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

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

管线参数

您可以使用参数将变量引入到管道定义中。您定义的参数可以在整个管线定义中参照。参数具有默认值,您可以在启动管道执行时通过指定参数值来覆盖该值。默认值必须是与参数类型匹配的实例。 步骤定义中使用的所有参数必须在管线定义中定义。Amazon SageMaker 模型构建管道支持以下参数类型:

  • ParameterString— 表示strPython 类型。

  • ParameterInteger— 表示intPython 类型。

  • ParameterFloat— 表示floatPython 类型。

参数采用以下格式:

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

下面是一个示例参数实现:

from sagemaker.workflow.parameters import ( ParameterInteger, ParameterString, ParameterFloat ) 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" ) )