Package software.amazon.awssdk.services.sagemaker.model

Interface TrainingJobDefinition.Builder

    • Method Detail

      • trainingInputMode

        TrainingJobDefinition.Builder trainingInputMode​(String trainingInputMode)
        Sets the value of the TrainingInputMode property for this object.
        Parameters:
        trainingInputMode - The new value for the TrainingInputMode property for this object.
        Returns:
        Returns a reference to this object so that method calls can be chained together.
        See Also:
        TrainingInputMode, TrainingInputMode

      • trainingInputMode

        TrainingJobDefinition.Builder trainingInputMode​(TrainingInputMode trainingInputMode)
        Sets the value of the TrainingInputMode property for this object.
        Parameters:
        trainingInputMode - The new value for the TrainingInputMode property for this object.
        Returns:
        Returns a reference to this object so that method calls can be chained together.
        See Also:
        TrainingInputMode, TrainingInputMode

      • hyperParameters

        TrainingJobDefinition.Builder hyperParameters​(Map<String,​String> hyperParameters)

        The hyperparameters used for the training job.

        Parameters:
        hyperParameters - The hyperparameters used for the training job.
        Returns:
        Returns a reference to this object so that method calls can be chained together.

      • inputDataConfig

        TrainingJobDefinition.Builder inputDataConfig​(Collection<Channel> inputDataConfig)

        An array of Channel objects, each of which specifies an input source.

        Parameters:
        inputDataConfig - An array of Channel objects, each of which specifies an input source.
        Returns:
        Returns a reference to this object so that method calls can be chained together.

      • inputDataConfig

        TrainingJobDefinition.Builder inputDataConfig​(Channel... inputDataConfig)

        An array of Channel objects, each of which specifies an input source.

        Parameters:
        inputDataConfig - An array of Channel objects, each of which specifies an input source.
        Returns:
        Returns a reference to this object so that method calls can be chained together.

      • inputDataConfig

        TrainingJobDefinition.Builder inputDataConfig​(Consumer<Channel.Builder>... inputDataConfig)

        An array of Channel objects, each of which specifies an input source.

        This is a convenience method that creates an instance of the Channel.Builder avoiding the need to create one manually via Channel.builder().

        When the Consumer completes, SdkBuilder.build() is called immediately and its result is passed to #inputDataConfig(List).

        Parameters:
        inputDataConfig - a consumer that will call methods on Channel.Builder
        Returns:
        Returns a reference to this object so that method calls can be chained together.
        See Also:
        #inputDataConfig(java.util.Collection)

      • outputDataConfig

        TrainingJobDefinition.Builder outputDataConfig​(OutputDataConfig outputDataConfig)

        the path to the S3 bucket where you want to store model artifacts. SageMaker creates subfolders for the artifacts.

        Parameters:
        outputDataConfig - the path to the S3 bucket where you want to store model artifacts. SageMaker creates subfolders for the artifacts.
        Returns:
        Returns a reference to this object so that method calls can be chained together.

      • resourceConfig

        TrainingJobDefinition.Builder resourceConfig​(ResourceConfig resourceConfig)

        The resources, including the ML compute instances and ML storage volumes, to use for model training.

        Parameters:
        resourceConfig - The resources, including the ML compute instances and ML storage volumes, to use for model training.
        Returns:
        Returns a reference to this object so that method calls can be chained together.

      • stoppingCondition

        TrainingJobDefinition.Builder stoppingCondition​(StoppingCondition stoppingCondition)

        Specifies a limit to how long a model training job can run. It also specifies how long a managed Spot training job has to complete. When the job reaches the time limit, SageMaker ends the training job. Use this API to cap model training costs.

        To stop a job, SageMaker sends the algorithm the SIGTERM signal, which delays job termination for 120 seconds. Algorithms can use this 120-second window to save the model artifacts.

        Parameters:
        stoppingCondition - Specifies a limit to how long a model training job can run. It also specifies how long a managed Spot training job has to complete. When the job reaches the time limit, SageMaker ends the training job. Use this API to cap model training costs.

        To stop a job, SageMaker sends the algorithm the SIGTERM signal, which delays job termination for 120 seconds. Algorithms can use this 120-second window to save the model artifacts.

        Returns:
        Returns a reference to this object so that method calls can be chained together.