software.amazon.awscdk.services.stepfunctions.tasks

Interface AlgorithmSpecification

All Known Implementing Classes:

AlgorithmSpecification.Jsii$Proxy

@Generated(value="jsii-pacmak/1.73.0 (build 6faeda3)",
           date="2023-01-25T18:29:11.119Z")
public interface AlgorithmSpecification

Specify the training algorithm and algorithm-specific metadata.

Example:

 SageMakerCreateTrainingJob.Builder.create(this, "TrainSagemaker")
         .trainingJobName(JsonPath.stringAt("$.JobName"))
         .algorithmSpecification(AlgorithmSpecification.builder()
                 .algorithmName("BlazingText")
                 .trainingInputMode(InputMode.FILE)
                 .build())
         .inputDataConfig(List.of(Channel.builder()
                 .channelName("train")
                 .dataSource(DataSource.builder()
                         .s3DataSource(S3DataSource.builder()
                                 .s3DataType(S3DataType.S3_PREFIX)
                                 .s3Location(S3Location.fromJsonExpression("$.S3Bucket"))
                                 .build())
                         .build())
                 .build()))
         .outputDataConfig(OutputDataConfig.builder()
                 .s3OutputLocation(S3Location.fromBucket(Bucket.fromBucketName(this, "Bucket", "mybucket"), "myoutputpath"))
                 .build())
         .resourceConfig(ResourceConfig.builder()
                 .instanceCount(1)
                 .instanceType(new InstanceType(JsonPath.stringAt("$.InstanceType")))
                 .volumeSize(Size.gibibytes(50))
                 .build()) // optional: default is 1 instance of EC2 `M4.XLarge` with `10GB` volume
         .stoppingCondition(StoppingCondition.builder()
                 .maxRuntime(Duration.hours(2))
                 .build())
         .build();
 

Nested Class Summary

Nested Classes

Modifier and Type	Interface and Description
static class	AlgorithmSpecification.Builder A builder for AlgorithmSpecification
static class	AlgorithmSpecification.Jsii$Proxy An implementation for AlgorithmSpecification

Method Summary

Modifier and Type	Method and Description
static AlgorithmSpecification.Builder	builder()
default java.lang.String	getAlgorithmName() Name of the algorithm resource to use for the training job.
default java.util.List<MetricDefinition>	getMetricDefinitions() List of metric definition objects.
default DockerImage	getTrainingImage() Registry path of the Docker image that contains the training algorithm.
default InputMode	getTrainingInputMode() Input mode that the algorithm supports.

Method Detail

getAlgorithmName

default java.lang.String getAlgorithmName()

Name of the algorithm resource to use for the training job.

This must be an algorithm resource that you created or subscribe to on AWS Marketplace. If you specify a value for this parameter, you can't specify a value for TrainingImage.

Default: - No algorithm is specified

getMetricDefinitions

default java.util.List<MetricDefinition> getMetricDefinitions()

List of metric definition objects.

Each object specifies the metric name and regular expressions used to parse algorithm logs.

Default: - No metrics

getTrainingImage

default DockerImage getTrainingImage()

Registry path of the Docker image that contains the training algorithm.

Default: - No Docker image is specified

getTrainingInputMode

default InputMode getTrainingInputMode()

Input mode that the algorithm supports.

Default: 'File' mode

builder

static AlgorithmSpecification.Builder builder()

Returns:

a AlgorithmSpecification.Builder of AlgorithmSpecification