software.amazon.awscdk.services.stepfunctions.tasks

Class SagemakerTrainTaskProps.Builder

java.lang.Object

software.amazon.awscdk.services.stepfunctions.tasks.SagemakerTrainTaskProps.Builder

Enclosing interface:

SagemakerTrainTaskProps

@Stability(value=Experimental)
public static final class SagemakerTrainTaskProps.Builder
extends Object

A builder for SagemakerTrainTaskProps

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
SagemakerTrainTaskProps.Builder	algorithmSpecification(AlgorithmSpecification algorithmSpecification) Sets the value of SagemakerTrainTaskProps.getAlgorithmSpecification()
SagemakerTrainTaskProps	build() Builds the configured instance.
SagemakerTrainTaskProps.Builder	hyperparameters(Map<String,Object> hyperparameters) Sets the value of SagemakerTrainTaskProps.getHyperparameters()
SagemakerTrainTaskProps.Builder	inputDataConfig(List<Channel> inputDataConfig) Sets the value of SagemakerTrainTaskProps.getInputDataConfig()
SagemakerTrainTaskProps.Builder	integrationPattern(ServiceIntegrationPattern integrationPattern) Sets the value of SagemakerTrainTaskProps.getIntegrationPattern()
SagemakerTrainTaskProps.Builder	outputDataConfig(OutputDataConfig outputDataConfig) Sets the value of SagemakerTrainTaskProps.getOutputDataConfig()
SagemakerTrainTaskProps.Builder	resourceConfig(ResourceConfig resourceConfig) Sets the value of SagemakerTrainTaskProps.getResourceConfig()
SagemakerTrainTaskProps.Builder	role(IRole role) Sets the value of SagemakerTrainTaskProps.getRole()
SagemakerTrainTaskProps.Builder	stoppingCondition(StoppingCondition stoppingCondition) Sets the value of SagemakerTrainTaskProps.getStoppingCondition()
SagemakerTrainTaskProps.Builder	tags(Map<String,String> tags) Sets the value of SagemakerTrainTaskProps.getTags()
SagemakerTrainTaskProps.Builder	trainingJobName(String trainingJobName) Sets the value of SagemakerTrainTaskProps.getTrainingJobName()
SagemakerTrainTaskProps.Builder	vpcConfig(VpcConfig vpcConfig) Sets the value of SagemakerTrainTaskProps.getVpcConfig()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Builder

public Builder()

Method Detail

algorithmSpecification

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder algorithmSpecification(AlgorithmSpecification algorithmSpecification)

Sets the value of SagemakerTrainTaskProps.getAlgorithmSpecification()

Parameters:

algorithmSpecification - Identifies the training algorithm to use. This parameter is required.

Returns:

this

inputDataConfig

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder inputDataConfig(List<Channel> inputDataConfig)

Sets the value of SagemakerTrainTaskProps.getInputDataConfig()

Parameters:

inputDataConfig - Describes the various datasets (e.g. train, validation, test) and the Amazon S3 location where stored. This parameter is required.

Returns:

this

outputDataConfig

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder outputDataConfig(OutputDataConfig outputDataConfig)

Sets the value of SagemakerTrainTaskProps.getOutputDataConfig()

Parameters:

outputDataConfig - Identifies the Amazon S3 location where you want Amazon SageMaker to save the results of model training. This parameter is required.

Returns:

this

trainingJobName

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder trainingJobName(String trainingJobName)

Sets the value of SagemakerTrainTaskProps.getTrainingJobName()

Parameters:

trainingJobName - Training Job Name. This parameter is required.

Returns:

this

hyperparameters

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder hyperparameters(Map<String,Object> hyperparameters)

Sets the value of SagemakerTrainTaskProps.getHyperparameters()

Parameters:

hyperparameters - Hyperparameters to be used for the train job.

Returns:

this

integrationPattern

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder integrationPattern(ServiceIntegrationPattern integrationPattern)

Sets the value of SagemakerTrainTaskProps.getIntegrationPattern()

Parameters:

integrationPattern - The service integration pattern indicates different ways to call SageMaker APIs. The valid value is either FIRE_AND_FORGET or SYNC.

Returns:

this

resourceConfig

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder resourceConfig(ResourceConfig resourceConfig)

Sets the value of SagemakerTrainTaskProps.getResourceConfig()

Parameters:

resourceConfig - Identifies the resources, ML compute instances, and ML storage volumes to deploy for model training.

Returns:

this

role

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder role(IRole role)

Sets the value of SagemakerTrainTaskProps.getRole()

Parameters:

role - Role for the Training Job. The role must be granted all necessary permissions for the SageMaker training job to be able to operate.

See https://docs.aws.amazon.com/fr_fr/sagemaker/latest/dg/sagemaker-roles.html#sagemaker-roles-createtrainingjob-perms

Returns:

this

stoppingCondition

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder stoppingCondition(StoppingCondition stoppingCondition)

Sets the value of SagemakerTrainTaskProps.getStoppingCondition()

Parameters:

stoppingCondition - Sets a time limit for training.

Returns:

this

tags

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder tags(Map<String,String> tags)

Sets the value of SagemakerTrainTaskProps.getTags()

Parameters:

tags - Tags to be applied to the train job.

Returns:

this

vpcConfig

@Stability(value=Experimental)
public SagemakerTrainTaskProps.Builder vpcConfig(VpcConfig vpcConfig)

Sets the value of SagemakerTrainTaskProps.getVpcConfig()

Parameters:

vpcConfig - Specifies the VPC that you want your training job to connect to.

Returns:

this

build

@Stability(value=Experimental)
public SagemakerTrainTaskProps build()

Builds the configured instance.

Returns:

a new instance of SagemakerTrainTaskProps

Throws:

NullPointerException - if any required attribute was not provided