Skip to content
v2.273.0

AlgorithmSpecification Struct ​

Specify the training algorithm and algorithm-specific metadata.

Example ​

ts
new tasks.SageMakerCreateTrainingJob(this, 'TrainSagemaker', {
  trainingJobName: sfn.JsonPath.stringAt('$.JobName'),
  algorithmSpecification: {
    algorithmName: 'BlazingText',
    trainingInputMode: tasks.InputMode.FILE,
  },
  inputDataConfig: [{
    channelName: 'train',
    dataSource: {
      s3DataSource: {
        s3DataType: tasks.S3DataType.S3_PREFIX,
        s3Location: tasks.S3Location.fromJsonExpression('$.S3Bucket'),
      },
    },
  }],
  outputDataConfig: {
    s3OutputLocation: tasks.S3Location.fromBucket(s3.Bucket.fromBucketName(this, 'Bucket', 'amzn-s3-demo-bucket'), 'myoutputpath'),
  },
  resourceConfig: {
    instanceCount: 1,
    instanceType: new ec2.InstanceType(sfn.JsonPath.stringAt('$.InstanceType')),
    volumeSize: Size.gibibytes(50),
  }, // optional: default is 1 instance of EC2 `M4.XLarge` with `10GB` volume
  stoppingCondition: {
    maxRuntime: Duration.hours(2),
  }, // optional: default is 1 hour
});

Properties ​

algorithmNamestringalgorithm_namestralgorithmNameStringAlgorithmNamestringAlgorithmName*string= No algorithm is specified#
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.

metricDefinitionsMetricDefinition[]metric_definitionstyping.List[MetricDefinition]metricDefinitionsList<MetricDefinition>MetricDefinitionsMetricDefinition[]MetricDefinitions*[]MetricDefinition= No metrics#
List of metric definition objects.

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

trainingImageDockerImagetraining_imageDockerImagetrainingImageDockerImageTrainingImageDockerImageTrainingImageDockerImage= No Docker image is specified#
Registry path of the Docker image that contains the training algorithm.
trainingInputModeInputModetraining_input_modeInputModetrainingInputModeInputModeTrainingInputModeInputModeTrainingInputModeInputMode= 'File' mode#
Input mode that the algorithm supports.