7#include <aws/core/utils/memory/stl/AWSMap.h>
8#include <aws/core/utils/memory/stl/AWSString.h>
9#include <aws/core/utils/memory/stl/AWSVector.h>
10#include <aws/sagemaker/SageMakerRequest.h>
11#include <aws/sagemaker/SageMaker_EXPORTS.h>
12#include <aws/sagemaker/model/AlgorithmSpecification.h>
13#include <aws/sagemaker/model/Channel.h>
14#include <aws/sagemaker/model/CheckpointConfig.h>
15#include <aws/sagemaker/model/DebugHookConfig.h>
16#include <aws/sagemaker/model/DebugRuleConfiguration.h>
17#include <aws/sagemaker/model/ExperimentConfig.h>
18#include <aws/sagemaker/model/InfraCheckConfig.h>
19#include <aws/sagemaker/model/MlflowConfig.h>
20#include <aws/sagemaker/model/ModelPackageConfig.h>
21#include <aws/sagemaker/model/OutputDataConfig.h>
22#include <aws/sagemaker/model/ProfilerConfig.h>
23#include <aws/sagemaker/model/ProfilerRuleConfiguration.h>
24#include <aws/sagemaker/model/RemoteDebugConfig.h>
25#include <aws/sagemaker/model/ResourceConfig.h>
26#include <aws/sagemaker/model/RetryStrategy.h>
27#include <aws/sagemaker/model/ServerlessJobConfig.h>
28#include <aws/sagemaker/model/SessionChainingConfig.h>
29#include <aws/sagemaker/model/StoppingCondition.h>
30#include <aws/sagemaker/model/Tag.h>
31#include <aws/sagemaker/model/TensorBoardOutputConfig.h>
32#include <aws/sagemaker/model/VpcConfig.h>
63 template <
typename TrainingJobNameT = Aws::String>
65 m_trainingJobNameHasBeenSet =
true;
66 m_trainingJobName = std::forward<TrainingJobNameT>(value);
68 template <
typename TrainingJobNameT = Aws::String>
92 template <
typename HyperParametersT = Aws::Map<Aws::String, Aws::String>>
94 m_hyperParametersHasBeenSet =
true;
95 m_hyperParameters = std::forward<HyperParametersT>(value);
97 template <
typename HyperParametersT = Aws::Map<Aws::String, Aws::String>>
102 template <
typename HyperParametersKeyT = Aws::String,
typename HyperParametersValueT = Aws::String>
104 m_hyperParametersHasBeenSet =
true;
105 m_hyperParameters.emplace(std::forward<HyperParametersKeyT>(key), std::forward<HyperParametersValueT>(value));
122 template <
typename AlgorithmSpecificationT = AlgorithmSpecification>
124 m_algorithmSpecificationHasBeenSet =
true;
125 m_algorithmSpecification = std::forward<AlgorithmSpecificationT>(value);
127 template <
typename AlgorithmSpecificationT = AlgorithmSpecification>
148 template <
typename RoleArnT = Aws::String>
150 m_roleArnHasBeenSet =
true;
151 m_roleArn = std::forward<RoleArnT>(value);
153 template <
typename RoleArnT = Aws::String>
178 template <
typename InputDataConfigT = Aws::Vector<Channel>>
180 m_inputDataConfigHasBeenSet =
true;
181 m_inputDataConfig = std::forward<InputDataConfigT>(value);
183 template <
typename InputDataConfigT = Aws::Vector<Channel>>
188 template <
typename InputDataConfigT = Channel>
190 m_inputDataConfigHasBeenSet =
true;
191 m_inputDataConfig.emplace_back(std::forward<InputDataConfigT>(value));
203 template <
typename OutputDataConfigT = OutputDataConfig>
205 m_outputDataConfigHasBeenSet =
true;
206 m_outputDataConfig = std::forward<OutputDataConfigT>(value);
208 template <
typename OutputDataConfigT = OutputDataConfig>
227 template <
typename ResourceConfigT = ResourceConfig>
229 m_resourceConfigHasBeenSet =
true;
230 m_resourceConfig = std::forward<ResourceConfigT>(value);
232 template <
typename ResourceConfigT = ResourceConfig>
251 template <
typename VpcConfigT = VpcConfig>
253 m_vpcConfigHasBeenSet =
true;
254 m_vpcConfig = std::forward<VpcConfigT>(value);
256 template <
typename VpcConfigT = VpcConfig>
275 template <
typename StoppingConditionT = StoppingCondition>
277 m_stoppingConditionHasBeenSet =
true;
278 m_stoppingCondition = std::forward<StoppingConditionT>(value);
280 template <
typename StoppingConditionT = StoppingCondition>
302 template <
typename TagsT = Aws::Vector<Tag>>
304 m_tagsHasBeenSet =
true;
305 m_tags = std::forward<TagsT>(value);
307 template <
typename TagsT = Aws::Vector<Tag>>
309 SetTags(std::forward<TagsT>(value));
312 template <
typename TagsT = Tag>
314 m_tagsHasBeenSet =
true;
315 m_tags.emplace_back(std::forward<TagsT>(value));
332 m_enableNetworkIsolationHasBeenSet =
true;
333 m_enableNetworkIsolation = value;
355 m_enableInterContainerTrafficEncryptionHasBeenSet =
true;
356 m_enableInterContainerTrafficEncryption = value;
379 m_enableManagedSpotTrainingHasBeenSet =
true;
380 m_enableManagedSpotTraining = value;
395 template <
typename Checkpo
intConfigT = Checkpo
intConfig>
397 m_checkpointConfigHasBeenSet =
true;
398 m_checkpointConfig = std::forward<CheckpointConfigT>(value);
400 template <
typename Checkpo
intConfigT = Checkpo
intConfig>
411 template <
typename DebugHookConfigT = DebugHookConfig>
413 m_debugHookConfigHasBeenSet =
true;
414 m_debugHookConfig = std::forward<DebugHookConfigT>(value);
416 template <
typename DebugHookConfigT = DebugHookConfig>
430 template <
typename DebugRuleConfigurationsT = Aws::Vector<DebugRuleConfiguration>>
432 m_debugRuleConfigurationsHasBeenSet =
true;
433 m_debugRuleConfigurations = std::forward<DebugRuleConfigurationsT>(value);
435 template <
typename DebugRuleConfigurationsT = Aws::Vector<DebugRuleConfiguration>>
440 template <
typename DebugRuleConfigurationsT = DebugRuleConfiguration>
442 m_debugRuleConfigurationsHasBeenSet =
true;
443 m_debugRuleConfigurations.emplace_back(std::forward<DebugRuleConfigurationsT>(value));
452 template <
typename TensorBoardOutputConfigT = TensorBoardOutputConfig>
454 m_tensorBoardOutputConfigHasBeenSet =
true;
455 m_tensorBoardOutputConfig = std::forward<TensorBoardOutputConfigT>(value);
457 template <
typename TensorBoardOutputConfigT = TensorBoardOutputConfig>
468 template <
typename ExperimentConfigT = ExperimentConfig>
470 m_experimentConfigHasBeenSet =
true;
471 m_experimentConfig = std::forward<ExperimentConfigT>(value);
473 template <
typename ExperimentConfigT = ExperimentConfig>
484 template <
typename ProfilerConfigT = ProfilerConfig>
486 m_profilerConfigHasBeenSet =
true;
487 m_profilerConfig = std::forward<ProfilerConfigT>(value);
489 template <
typename ProfilerConfigT = ProfilerConfig>
503 template <
typename ProfilerRuleConfigurationsT = Aws::Vector<ProfilerRuleConfiguration>>
505 m_profilerRuleConfigurationsHasBeenSet =
true;
506 m_profilerRuleConfigurations = std::forward<ProfilerRuleConfigurationsT>(value);
508 template <
typename ProfilerRuleConfigurationsT = Aws::Vector<ProfilerRuleConfiguration>>
513 template <
typename ProfilerRuleConfigurationsT = ProfilerRuleConfiguration>
515 m_profilerRuleConfigurationsHasBeenSet =
true;
516 m_profilerRuleConfigurations.emplace_back(std::forward<ProfilerRuleConfigurationsT>(value));
533 template <
typename EnvironmentT = Aws::Map<Aws::String, Aws::String>>
535 m_environmentHasBeenSet =
true;
536 m_environment = std::forward<EnvironmentT>(value);
538 template <
typename EnvironmentT = Aws::Map<Aws::String, Aws::String>>
543 template <
typename EnvironmentKeyT = Aws::String,
typename EnvironmentValueT = Aws::String>
545 m_environmentHasBeenSet =
true;
546 m_environment.emplace(std::forward<EnvironmentKeyT>(key), std::forward<EnvironmentValueT>(value));
558 template <
typename RetryStrategyT = RetryStrategy>
560 m_retryStrategyHasBeenSet =
true;
561 m_retryStrategy = std::forward<RetryStrategyT>(value);
563 template <
typename RetryStrategyT = RetryStrategy>
580 template <
typename RemoteDebugConfigT = RemoteDebugConfig>
582 m_remoteDebugConfigHasBeenSet =
true;
583 m_remoteDebugConfig = std::forward<RemoteDebugConfigT>(value);
585 template <
typename RemoteDebugConfigT = RemoteDebugConfig>
599 template <
typename InfraCheckConfigT = InfraCheckConfig>
601 m_infraCheckConfigHasBeenSet =
true;
602 m_infraCheckConfig = std::forward<InfraCheckConfigT>(value);
604 template <
typename InfraCheckConfigT = InfraCheckConfig>
618 template <
typename SessionChainingConfigT = SessionChainingConfig>
620 m_sessionChainingConfigHasBeenSet =
true;
621 m_sessionChainingConfig = std::forward<SessionChainingConfigT>(value);
623 template <
typename SessionChainingConfigT = SessionChainingConfig>
636 template <
typename ServerlessJobConfigT = ServerlessJobConfig>
638 m_serverlessJobConfigHasBeenSet =
true;
639 m_serverlessJobConfig = std::forward<ServerlessJobConfigT>(value);
641 template <
typename ServerlessJobConfigT = ServerlessJobConfig>
654 template <
typename MlflowConfigT = MlflowConfig>
656 m_mlflowConfigHasBeenSet =
true;
657 m_mlflowConfig = std::forward<MlflowConfigT>(value);
659 template <
typename MlflowConfigT = MlflowConfig>
672 template <
typename ModelPackageConfigT = ModelPackageConfig>
674 m_modelPackageConfigHasBeenSet =
true;
675 m_modelPackageConfig = std::forward<ModelPackageConfigT>(value);
677 template <
typename ModelPackageConfigT = ModelPackageConfig>
704 bool m_enableNetworkIsolation{
false};
706 bool m_enableInterContainerTrafficEncryption{
false};
708 bool m_enableManagedSpotTraining{
false};
710 CheckpointConfig m_checkpointConfig;
712 DebugHookConfig m_debugHookConfig;
716 TensorBoardOutputConfig m_tensorBoardOutputConfig;
718 ExperimentConfig m_experimentConfig;
720 ProfilerConfig m_profilerConfig;
726 RetryStrategy m_retryStrategy;
728 RemoteDebugConfig m_remoteDebugConfig;
730 InfraCheckConfig m_infraCheckConfig;
732 SessionChainingConfig m_sessionChainingConfig;
734 ServerlessJobConfig m_serverlessJobConfig;
736 MlflowConfig m_mlflowConfig;
738 ModelPackageConfig m_modelPackageConfig;
739 bool m_trainingJobNameHasBeenSet =
false;
740 bool m_hyperParametersHasBeenSet =
false;
741 bool m_algorithmSpecificationHasBeenSet =
false;
742 bool m_roleArnHasBeenSet =
false;
743 bool m_inputDataConfigHasBeenSet =
false;
744 bool m_outputDataConfigHasBeenSet =
false;
745 bool m_resourceConfigHasBeenSet =
false;
746 bool m_vpcConfigHasBeenSet =
false;
747 bool m_stoppingConditionHasBeenSet =
false;
748 bool m_tagsHasBeenSet =
false;
749 bool m_enableNetworkIsolationHasBeenSet =
false;
750 bool m_enableInterContainerTrafficEncryptionHasBeenSet =
false;
751 bool m_enableManagedSpotTrainingHasBeenSet =
false;
752 bool m_checkpointConfigHasBeenSet =
false;
753 bool m_debugHookConfigHasBeenSet =
false;
754 bool m_debugRuleConfigurationsHasBeenSet =
false;
755 bool m_tensorBoardOutputConfigHasBeenSet =
false;
756 bool m_experimentConfigHasBeenSet =
false;
757 bool m_profilerConfigHasBeenSet =
false;
758 bool m_profilerRuleConfigurationsHasBeenSet =
false;
759 bool m_environmentHasBeenSet =
false;
760 bool m_retryStrategyHasBeenSet =
false;
761 bool m_remoteDebugConfigHasBeenSet =
false;
762 bool m_infraCheckConfigHasBeenSet =
false;
763 bool m_sessionChainingConfigHasBeenSet =
false;
764 bool m_serverlessJobConfigHasBeenSet =
false;
765 bool m_mlflowConfigHasBeenSet =
false;
766 bool m_modelPackageConfigHasBeenSet =
false;
const StoppingCondition & GetStoppingCondition() const
virtual const char * GetServiceRequestName() const override
bool ServerlessJobConfigHasBeenSet() const
bool VpcConfigHasBeenSet() const
void SetInfraCheckConfig(InfraCheckConfigT &&value)
void SetEnableManagedSpotTraining(bool value)
CreateTrainingJobRequest & WithEnableInterContainerTrafficEncryption(bool value)
CreateTrainingJobRequest & WithResourceConfig(ResourceConfigT &&value)
const Aws::Vector< DebugRuleConfiguration > & GetDebugRuleConfigurations() const
const ResourceConfig & GetResourceConfig() const
bool ResourceConfigHasBeenSet() const
CreateTrainingJobRequest & WithSessionChainingConfig(SessionChainingConfigT &&value)
bool ProfilerRuleConfigurationsHasBeenSet() const
AWS_SAGEMAKER_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override
void SetModelPackageConfig(ModelPackageConfigT &&value)
void SetRetryStrategy(RetryStrategyT &&value)
bool GetEnableInterContainerTrafficEncryption() const
const ModelPackageConfig & GetModelPackageConfig() const
const InfraCheckConfig & GetInfraCheckConfig() const
void SetMlflowConfig(MlflowConfigT &&value)
CreateTrainingJobRequest & WithVpcConfig(VpcConfigT &&value)
const SessionChainingConfig & GetSessionChainingConfig() const
CreateTrainingJobRequest & WithEnableNetworkIsolation(bool value)
const Aws::Map< Aws::String, Aws::String > & GetHyperParameters() const
CreateTrainingJobRequest & WithCheckpointConfig(CheckpointConfigT &&value)
CreateTrainingJobRequest & WithModelPackageConfig(ModelPackageConfigT &&value)
CreateTrainingJobRequest & WithDebugRuleConfigurations(DebugRuleConfigurationsT &&value)
AWS_SAGEMAKER_API CreateTrainingJobRequest()=default
const Aws::Vector< Channel > & GetInputDataConfig() const
void SetServerlessJobConfig(ServerlessJobConfigT &&value)
CreateTrainingJobRequest & WithRemoteDebugConfig(RemoteDebugConfigT &&value)
bool TensorBoardOutputConfigHasBeenSet() const
bool SessionChainingConfigHasBeenSet() const
bool OutputDataConfigHasBeenSet() const
CreateTrainingJobRequest & WithTrainingJobName(TrainingJobNameT &&value)
CreateTrainingJobRequest & WithServerlessJobConfig(ServerlessJobConfigT &&value)
void SetEnvironment(EnvironmentT &&value)
bool MlflowConfigHasBeenSet() const
CreateTrainingJobRequest & AddProfilerRuleConfigurations(ProfilerRuleConfigurationsT &&value)
const ProfilerConfig & GetProfilerConfig() const
const ExperimentConfig & GetExperimentConfig() const
void SetProfilerConfig(ProfilerConfigT &&value)
const CheckpointConfig & GetCheckpointConfig() const
bool InfraCheckConfigHasBeenSet() const
bool ProfilerConfigHasBeenSet() const
CreateTrainingJobRequest & WithTags(TagsT &&value)
CreateTrainingJobRequest & WithEnableManagedSpotTraining(bool value)
void SetDebugHookConfig(DebugHookConfigT &&value)
CreateTrainingJobRequest & AddEnvironment(EnvironmentKeyT &&key, EnvironmentValueT &&value)
void SetHyperParameters(HyperParametersT &&value)
const AlgorithmSpecification & GetAlgorithmSpecification() const
CreateTrainingJobRequest & AddInputDataConfig(InputDataConfigT &&value)
void SetEnableInterContainerTrafficEncryption(bool value)
const TensorBoardOutputConfig & GetTensorBoardOutputConfig() const
bool DebugHookConfigHasBeenSet() const
bool EnvironmentHasBeenSet() const
CreateTrainingJobRequest & WithExperimentConfig(ExperimentConfigT &&value)
bool EnableManagedSpotTrainingHasBeenSet() const
bool GetEnableManagedSpotTraining() const
bool AlgorithmSpecificationHasBeenSet() const
const DebugHookConfig & GetDebugHookConfig() const
const Aws::Vector< Tag > & GetTags() const
CreateTrainingJobRequest & WithStoppingCondition(StoppingConditionT &&value)
void SetOutputDataConfig(OutputDataConfigT &&value)
CreateTrainingJobRequest & AddHyperParameters(HyperParametersKeyT &&key, HyperParametersValueT &&value)
const Aws::Map< Aws::String, Aws::String > & GetEnvironment() const
CreateTrainingJobRequest & WithRetryStrategy(RetryStrategyT &&value)
CreateTrainingJobRequest & WithTensorBoardOutputConfig(TensorBoardOutputConfigT &&value)
CreateTrainingJobRequest & WithAlgorithmSpecification(AlgorithmSpecificationT &&value)
const Aws::String & GetTrainingJobName() const
bool GetEnableNetworkIsolation() const
void SetDebugRuleConfigurations(DebugRuleConfigurationsT &&value)
CreateTrainingJobRequest & AddTags(TagsT &&value)
void SetRemoteDebugConfig(RemoteDebugConfigT &&value)
bool DebugRuleConfigurationsHasBeenSet() const
const ServerlessJobConfig & GetServerlessJobConfig() const
void SetCheckpointConfig(CheckpointConfigT &&value)
void SetTrainingJobName(TrainingJobNameT &&value)
const OutputDataConfig & GetOutputDataConfig() const
CreateTrainingJobRequest & AddDebugRuleConfigurations(DebugRuleConfigurationsT &&value)
void SetProfilerRuleConfigurations(ProfilerRuleConfigurationsT &&value)
bool TagsHasBeenSet() const
void SetInputDataConfig(InputDataConfigT &&value)
const RetryStrategy & GetRetryStrategy() const
void SetTags(TagsT &&value)
bool ModelPackageConfigHasBeenSet() const
bool InputDataConfigHasBeenSet() const
bool RemoteDebugConfigHasBeenSet() const
bool StoppingConditionHasBeenSet() const
CreateTrainingJobRequest & WithHyperParameters(HyperParametersT &&value)
void SetEnableNetworkIsolation(bool value)
void SetSessionChainingConfig(SessionChainingConfigT &&value)
CreateTrainingJobRequest & WithDebugHookConfig(DebugHookConfigT &&value)
bool CheckpointConfigHasBeenSet() const
bool RoleArnHasBeenSet() const
bool EnableNetworkIsolationHasBeenSet() const
const MlflowConfig & GetMlflowConfig() const
void SetExperimentConfig(ExperimentConfigT &&value)
const Aws::String & GetRoleArn() const
void SetTensorBoardOutputConfig(TensorBoardOutputConfigT &&value)
CreateTrainingJobRequest & WithMlflowConfig(MlflowConfigT &&value)
CreateTrainingJobRequest & WithInfraCheckConfig(InfraCheckConfigT &&value)
void SetAlgorithmSpecification(AlgorithmSpecificationT &&value)
bool RetryStrategyHasBeenSet() const
void SetRoleArn(RoleArnT &&value)
const Aws::Vector< ProfilerRuleConfiguration > & GetProfilerRuleConfigurations() const
CreateTrainingJobRequest & WithProfilerRuleConfigurations(ProfilerRuleConfigurationsT &&value)
CreateTrainingJobRequest & WithInputDataConfig(InputDataConfigT &&value)
void SetStoppingCondition(StoppingConditionT &&value)
CreateTrainingJobRequest & WithEnvironment(EnvironmentT &&value)
bool HyperParametersHasBeenSet() const
void SetResourceConfig(ResourceConfigT &&value)
const VpcConfig & GetVpcConfig() const
AWS_SAGEMAKER_API Aws::String SerializePayload() const override
CreateTrainingJobRequest & WithProfilerConfig(ProfilerConfigT &&value)
bool EnableInterContainerTrafficEncryptionHasBeenSet() const
const RemoteDebugConfig & GetRemoteDebugConfig() const
void SetVpcConfig(VpcConfigT &&value)
bool TrainingJobNameHasBeenSet() const
bool ExperimentConfigHasBeenSet() const
CreateTrainingJobRequest & WithOutputDataConfig(OutputDataConfigT &&value)
CreateTrainingJobRequest & WithRoleArn(RoleArnT &&value)
Aws::Map< Aws::String, Aws::String > HeaderValueCollection
std::map< K, V, std::less< K >, Aws::Allocator< std::pair< const K, V > > > Map
std::basic_string< char, std::char_traits< char >, Aws::Allocator< char > > String
std::vector< T, Aws::Allocator< T > > Vector