7#include <aws/sagemaker/SageMaker_EXPORTS.h>
8#include <aws/sagemaker/SageMakerRequest.h>
9#include <aws/core/utils/memory/stl/AWSString.h>
10#include <aws/core/utils/memory/stl/AWSMap.h>
11#include <aws/sagemaker/model/AlgorithmSpecification.h>
12#include <aws/core/utils/memory/stl/AWSVector.h>
13#include <aws/sagemaker/model/OutputDataConfig.h>
14#include <aws/sagemaker/model/ResourceConfig.h>
15#include <aws/sagemaker/model/VpcConfig.h>
16#include <aws/sagemaker/model/StoppingCondition.h>
17#include <aws/sagemaker/model/CheckpointConfig.h>
18#include <aws/sagemaker/model/DebugHookConfig.h>
19#include <aws/sagemaker/model/TensorBoardOutputConfig.h>
20#include <aws/sagemaker/model/ExperimentConfig.h>
21#include <aws/sagemaker/model/ProfilerConfig.h>
22#include <aws/sagemaker/model/RetryStrategy.h>
23#include <aws/sagemaker/model/RemoteDebugConfig.h>
24#include <aws/sagemaker/model/InfraCheckConfig.h>
25#include <aws/sagemaker/model/SessionChainingConfig.h>
26#include <aws/sagemaker/model/Channel.h>
27#include <aws/sagemaker/model/Tag.h>
28#include <aws/sagemaker/model/DebugRuleConfiguration.h>
29#include <aws/sagemaker/model/ProfilerRuleConfiguration.h>
66 inline void SetTrainingJobName(
const char* value) { m_trainingJobNameHasBeenSet =
true; m_trainingJobName.assign(value); }
135 inline void SetRoleArn(
const char* value) { m_roleArnHasBeenSet =
true; m_roleArn.assign(value); }
461 bool m_trainingJobNameHasBeenSet =
false;
464 bool m_hyperParametersHasBeenSet =
false;
467 bool m_algorithmSpecificationHasBeenSet =
false;
470 bool m_roleArnHasBeenSet =
false;
473 bool m_inputDataConfigHasBeenSet =
false;
476 bool m_outputDataConfigHasBeenSet =
false;
479 bool m_resourceConfigHasBeenSet =
false;
482 bool m_vpcConfigHasBeenSet =
false;
485 bool m_stoppingConditionHasBeenSet =
false;
488 bool m_tagsHasBeenSet =
false;
490 bool m_enableNetworkIsolation;
491 bool m_enableNetworkIsolationHasBeenSet =
false;
493 bool m_enableInterContainerTrafficEncryption;
494 bool m_enableInterContainerTrafficEncryptionHasBeenSet =
false;
496 bool m_enableManagedSpotTraining;
497 bool m_enableManagedSpotTrainingHasBeenSet =
false;
500 bool m_checkpointConfigHasBeenSet =
false;
503 bool m_debugHookConfigHasBeenSet =
false;
506 bool m_debugRuleConfigurationsHasBeenSet =
false;
509 bool m_tensorBoardOutputConfigHasBeenSet =
false;
512 bool m_experimentConfigHasBeenSet =
false;
515 bool m_profilerConfigHasBeenSet =
false;
518 bool m_profilerRuleConfigurationsHasBeenSet =
false;
521 bool m_environmentHasBeenSet =
false;
524 bool m_retryStrategyHasBeenSet =
false;
527 bool m_remoteDebugConfigHasBeenSet =
false;
530 bool m_infraCheckConfigHasBeenSet =
false;
533 bool m_sessionChainingConfigHasBeenSet =
false;
const StoppingCondition & GetStoppingCondition() const
virtual const char * GetServiceRequestName() const override
CreateTrainingJobRequest & WithRoleArn(Aws::String &&value)
CreateTrainingJobRequest & AddDebugRuleConfigurations(DebugRuleConfiguration &&value)
CreateTrainingJobRequest & AddProfilerRuleConfigurations(const ProfilerRuleConfiguration &value)
bool VpcConfigHasBeenSet() const
void SetInputDataConfig(const Aws::Vector< Channel > &value)
CreateTrainingJobRequest & AddHyperParameters(Aws::String &&key, const Aws::String &value)
void SetEnableManagedSpotTraining(bool value)
void SetCheckpointConfig(CheckpointConfig &&value)
CreateTrainingJobRequest & WithEnableInterContainerTrafficEncryption(bool value)
CreateTrainingJobRequest & WithTrainingJobName(const Aws::String &value)
CreateTrainingJobRequest & WithRoleArn(const char *value)
void SetHyperParameters(const Aws::Map< Aws::String, Aws::String > &value)
AWS_SAGEMAKER_API CreateTrainingJobRequest()
void SetOutputDataConfig(OutputDataConfig &&value)
const Aws::Vector< DebugRuleConfiguration > & GetDebugRuleConfigurations() const
const ResourceConfig & GetResourceConfig() const
bool ResourceConfigHasBeenSet() const
CreateTrainingJobRequest & WithCheckpointConfig(const CheckpointConfig &value)
void SetSessionChainingConfig(const SessionChainingConfig &value)
bool ProfilerRuleConfigurationsHasBeenSet() const
AWS_SAGEMAKER_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override
void SetDebugHookConfig(const DebugHookConfig &value)
bool GetEnableInterContainerTrafficEncryption() const
const InfraCheckConfig & GetInfraCheckConfig() const
CreateTrainingJobRequest & WithStoppingCondition(StoppingCondition &&value)
void SetRemoteDebugConfig(const RemoteDebugConfig &value)
void SetExperimentConfig(ExperimentConfig &&value)
const SessionChainingConfig & GetSessionChainingConfig() const
void SetStoppingCondition(StoppingCondition &&value)
CreateTrainingJobRequest & WithEnableNetworkIsolation(bool value)
const Aws::Map< Aws::String, Aws::String > & GetHyperParameters() const
void SetTrainingJobName(Aws::String &&value)
void SetProfilerRuleConfigurations(Aws::Vector< ProfilerRuleConfiguration > &&value)
CreateTrainingJobRequest & WithOutputDataConfig(OutputDataConfig &&value)
void SetEnvironment(const Aws::Map< Aws::String, Aws::String > &value)
CreateTrainingJobRequest & AddTags(Tag &&value)
CreateTrainingJobRequest & WithHyperParameters(const Aws::Map< Aws::String, Aws::String > &value)
CreateTrainingJobRequest & WithRemoteDebugConfig(RemoteDebugConfig &&value)
CreateTrainingJobRequest & AddHyperParameters(Aws::String &&key, const char *value)
CreateTrainingJobRequest & WithCheckpointConfig(CheckpointConfig &&value)
const Aws::Vector< Channel > & GetInputDataConfig() const
CreateTrainingJobRequest & AddEnvironment(Aws::String &&key, const Aws::String &value)
bool TensorBoardOutputConfigHasBeenSet() const
CreateTrainingJobRequest & WithVpcConfig(const VpcConfig &value)
bool SessionChainingConfigHasBeenSet() const
bool OutputDataConfigHasBeenSet() const
CreateTrainingJobRequest & WithProfilerConfig(ProfilerConfig &&value)
CreateTrainingJobRequest & AddHyperParameters(const Aws::String &key, Aws::String &&value)
CreateTrainingJobRequest & WithInputDataConfig(const Aws::Vector< Channel > &value)
CreateTrainingJobRequest & WithAlgorithmSpecification(const AlgorithmSpecification &value)
void SetRetryStrategy(const RetryStrategy &value)
CreateTrainingJobRequest & AddEnvironment(Aws::String &&key, Aws::String &&value)
CreateTrainingJobRequest & AddEnvironment(const Aws::String &key, const Aws::String &value)
CreateTrainingJobRequest & WithRoleArn(const Aws::String &value)
CreateTrainingJobRequest & AddHyperParameters(const char *key, Aws::String &&value)
void SetRemoteDebugConfig(RemoteDebugConfig &&value)
CreateTrainingJobRequest & WithDebugRuleConfigurations(const Aws::Vector< DebugRuleConfiguration > &value)
const ProfilerConfig & GetProfilerConfig() const
CreateTrainingJobRequest & AddHyperParameters(Aws::String &&key, Aws::String &&value)
const ExperimentConfig & GetExperimentConfig() const
CreateTrainingJobRequest & WithResourceConfig(ResourceConfig &&value)
void SetRoleArn(const char *value)
const CheckpointConfig & GetCheckpointConfig() const
bool InfraCheckConfigHasBeenSet() const
bool ProfilerConfigHasBeenSet() const
CreateTrainingJobRequest & WithEnableManagedSpotTraining(bool value)
void SetTrainingJobName(const char *value)
CreateTrainingJobRequest & WithTensorBoardOutputConfig(TensorBoardOutputConfig &&value)
const AlgorithmSpecification & GetAlgorithmSpecification() const
void SetEnableInterContainerTrafficEncryption(bool value)
const TensorBoardOutputConfig & GetTensorBoardOutputConfig() const
void SetResourceConfig(ResourceConfig &&value)
bool DebugHookConfigHasBeenSet() const
CreateTrainingJobRequest & WithAlgorithmSpecification(AlgorithmSpecification &&value)
CreateTrainingJobRequest & WithRemoteDebugConfig(const RemoteDebugConfig &value)
CreateTrainingJobRequest & AddDebugRuleConfigurations(const DebugRuleConfiguration &value)
bool EnvironmentHasBeenSet() const
bool EnableManagedSpotTrainingHasBeenSet() const
bool GetEnableManagedSpotTraining() const
CreateTrainingJobRequest & AddEnvironment(const char *key, Aws::String &&value)
CreateTrainingJobRequest & WithInfraCheckConfig(const InfraCheckConfig &value)
bool AlgorithmSpecificationHasBeenSet() const
const DebugHookConfig & GetDebugHookConfig() const
const Aws::Vector< Tag > & GetTags() const
void SetCheckpointConfig(const CheckpointConfig &value)
void SetVpcConfig(VpcConfig &&value)
CreateTrainingJobRequest & WithTags(Aws::Vector< Tag > &&value)
void SetRetryStrategy(RetryStrategy &&value)
CreateTrainingJobRequest & WithTags(const Aws::Vector< Tag > &value)
void SetAlgorithmSpecification(AlgorithmSpecification &&value)
const Aws::Map< Aws::String, Aws::String > & GetEnvironment() const
void SetRoleArn(const Aws::String &value)
void SetTensorBoardOutputConfig(const TensorBoardOutputConfig &value)
const Aws::String & GetTrainingJobName() const
bool GetEnableNetworkIsolation() const
void SetStoppingCondition(const StoppingCondition &value)
CreateTrainingJobRequest & WithStoppingCondition(const StoppingCondition &value)
void SetProfilerConfig(ProfilerConfig &&value)
bool DebugRuleConfigurationsHasBeenSet() const
void SetExperimentConfig(const ExperimentConfig &value)
CreateTrainingJobRequest & WithRetryStrategy(const RetryStrategy &value)
CreateTrainingJobRequest & WithProfilerRuleConfigurations(const Aws::Vector< ProfilerRuleConfiguration > &value)
const OutputDataConfig & GetOutputDataConfig() const
CreateTrainingJobRequest & WithDebugHookConfig(DebugHookConfig &&value)
CreateTrainingJobRequest & WithInfraCheckConfig(InfraCheckConfig &&value)
void SetOutputDataConfig(const OutputDataConfig &value)
CreateTrainingJobRequest & WithExperimentConfig(const ExperimentConfig &value)
CreateTrainingJobRequest & WithVpcConfig(VpcConfig &&value)
bool TagsHasBeenSet() const
const RetryStrategy & GetRetryStrategy() const
CreateTrainingJobRequest & WithDebugHookConfig(const DebugHookConfig &value)
bool InputDataConfigHasBeenSet() const
bool RemoteDebugConfigHasBeenSet() const
bool StoppingConditionHasBeenSet() const
CreateTrainingJobRequest & AddHyperParameters(const Aws::String &key, const Aws::String &value)
void SetEnableNetworkIsolation(bool value)
void SetRoleArn(Aws::String &&value)
void SetTags(const Aws::Vector< Tag > &value)
void SetSessionChainingConfig(SessionChainingConfig &&value)
bool CheckpointConfigHasBeenSet() const
bool RoleArnHasBeenSet() const
bool EnableNetworkIsolationHasBeenSet() const
void SetEnvironment(Aws::Map< Aws::String, Aws::String > &&value)
void SetInfraCheckConfig(const InfraCheckConfig &value)
CreateTrainingJobRequest & WithRetryStrategy(RetryStrategy &&value)
void SetProfilerConfig(const ProfilerConfig &value)
void SetDebugRuleConfigurations(Aws::Vector< DebugRuleConfiguration > &&value)
void SetDebugHookConfig(DebugHookConfig &&value)
const Aws::String & GetRoleArn() const
void SetAlgorithmSpecification(const AlgorithmSpecification &value)
CreateTrainingJobRequest & WithTrainingJobName(Aws::String &&value)
CreateTrainingJobRequest & AddEnvironment(Aws::String &&key, const char *value)
CreateTrainingJobRequest & WithResourceConfig(const ResourceConfig &value)
void SetDebugRuleConfigurations(const Aws::Vector< DebugRuleConfiguration > &value)
bool RetryStrategyHasBeenSet() const
const Aws::Vector< ProfilerRuleConfiguration > & GetProfilerRuleConfigurations() const
CreateTrainingJobRequest & AddHyperParameters(const char *key, const char *value)
void SetTrainingJobName(const Aws::String &value)
CreateTrainingJobRequest & WithSessionChainingConfig(SessionChainingConfig &&value)
void SetTags(Aws::Vector< Tag > &&value)
CreateTrainingJobRequest & AddProfilerRuleConfigurations(ProfilerRuleConfiguration &&value)
CreateTrainingJobRequest & AddInputDataConfig(const Channel &value)
bool HyperParametersHasBeenSet() const
void SetVpcConfig(const VpcConfig &value)
CreateTrainingJobRequest & WithInputDataConfig(Aws::Vector< Channel > &&value)
CreateTrainingJobRequest & WithProfilerConfig(const ProfilerConfig &value)
CreateTrainingJobRequest & WithHyperParameters(Aws::Map< Aws::String, Aws::String > &&value)
CreateTrainingJobRequest & WithExperimentConfig(ExperimentConfig &&value)
CreateTrainingJobRequest & WithEnvironment(Aws::Map< Aws::String, Aws::String > &&value)
void SetTensorBoardOutputConfig(TensorBoardOutputConfig &&value)
const VpcConfig & GetVpcConfig() const
CreateTrainingJobRequest & WithProfilerRuleConfigurations(Aws::Vector< ProfilerRuleConfiguration > &&value)
AWS_SAGEMAKER_API Aws::String SerializePayload() const override
void SetResourceConfig(const ResourceConfig &value)
bool EnableInterContainerTrafficEncryptionHasBeenSet() const
const RemoteDebugConfig & GetRemoteDebugConfig() const
CreateTrainingJobRequest & WithEnvironment(const Aws::Map< Aws::String, Aws::String > &value)
CreateTrainingJobRequest & AddEnvironment(const Aws::String &key, Aws::String &&value)
CreateTrainingJobRequest & WithOutputDataConfig(const OutputDataConfig &value)
bool TrainingJobNameHasBeenSet() const
CreateTrainingJobRequest & WithDebugRuleConfigurations(Aws::Vector< DebugRuleConfiguration > &&value)
void SetInfraCheckConfig(InfraCheckConfig &&value)
CreateTrainingJobRequest & WithTrainingJobName(const char *value)
void SetProfilerRuleConfigurations(const Aws::Vector< ProfilerRuleConfiguration > &value)
bool ExperimentConfigHasBeenSet() const
void SetHyperParameters(Aws::Map< Aws::String, Aws::String > &&value)
void SetInputDataConfig(Aws::Vector< Channel > &&value)
CreateTrainingJobRequest & AddEnvironment(const char *key, const char *value)
CreateTrainingJobRequest & AddTags(const Tag &value)
CreateTrainingJobRequest & WithSessionChainingConfig(const SessionChainingConfig &value)
CreateTrainingJobRequest & WithTensorBoardOutputConfig(const TensorBoardOutputConfig &value)
CreateTrainingJobRequest & AddInputDataConfig(Channel &&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