HyperParameterTuningJobConfig.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/HyperParameterTuningJobObjective.h>
#include <aws/sagemaker/model/HyperParameterTuningJobStrategyConfig.h>
#include <aws/sagemaker/model/HyperParameterTuningJobStrategyType.h>
#include <aws/sagemaker/model/ParameterRanges.h>
#include <aws/sagemaker/model/ResourceLimits.h>
#include <aws/sagemaker/model/TrainingJobEarlyStoppingType.h>
#include <aws/sagemaker/model/TuningJobCompletionCriteria.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class HyperParameterTuningJobConfig {
 public:
  AWS_SAGEMAKER_API HyperParameterTuningJobConfig() = default;
  AWS_SAGEMAKER_API HyperParameterTuningJobConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API HyperParameterTuningJobConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline HyperParameterTuningJobStrategyType GetStrategy() const { return m_strategy; }
  inline bool StrategyHasBeenSet() const { return m_strategyHasBeenSet; }
  inline void SetStrategy(HyperParameterTuningJobStrategyType value) {
    m_strategyHasBeenSet = true;
    m_strategy = value;
  }
  inline HyperParameterTuningJobConfig& WithStrategy(HyperParameterTuningJobStrategyType value) {
    SetStrategy(value);
    return *this;
  }
 
 
  inline const HyperParameterTuningJobStrategyConfig& GetStrategyConfig() const { return m_strategyConfig; }
  inline bool StrategyConfigHasBeenSet() const { return m_strategyConfigHasBeenSet; }
  template <typename StrategyConfigT = HyperParameterTuningJobStrategyConfig>
  void SetStrategyConfig(StrategyConfigT&& value) {
    m_strategyConfigHasBeenSet = true;
    m_strategyConfig = std::forward<StrategyConfigT>(value);
  }
  template <typename StrategyConfigT = HyperParameterTuningJobStrategyConfig>
  HyperParameterTuningJobConfig& WithStrategyConfig(StrategyConfigT&& value) {
    SetStrategyConfig(std::forward<StrategyConfigT>(value));
    return *this;
  }
 
 
  inline const HyperParameterTuningJobObjective& GetHyperParameterTuningJobObjective() const { return m_hyperParameterTuningJobObjective; }
  inline bool HyperParameterTuningJobObjectiveHasBeenSet() const { return m_hyperParameterTuningJobObjectiveHasBeenSet; }
  template <typename HyperParameterTuningJobObjectiveT = HyperParameterTuningJobObjective>
  void SetHyperParameterTuningJobObjective(HyperParameterTuningJobObjectiveT&& value) {
    m_hyperParameterTuningJobObjectiveHasBeenSet = true;
    m_hyperParameterTuningJobObjective = std::forward<HyperParameterTuningJobObjectiveT>(value);
  }
  template <typename HyperParameterTuningJobObjectiveT = HyperParameterTuningJobObjective>
  HyperParameterTuningJobConfig& WithHyperParameterTuningJobObjective(HyperParameterTuningJobObjectiveT&& value) {
    SetHyperParameterTuningJobObjective(std::forward<HyperParameterTuningJobObjectiveT>(value));
    return *this;
  }
 
 
  inline const ResourceLimits& GetResourceLimits() const { return m_resourceLimits; }
  inline bool ResourceLimitsHasBeenSet() const { return m_resourceLimitsHasBeenSet; }
  template <typename ResourceLimitsT = ResourceLimits>
  void SetResourceLimits(ResourceLimitsT&& value) {
    m_resourceLimitsHasBeenSet = true;
    m_resourceLimits = std::forward<ResourceLimitsT>(value);
  }
  template <typename ResourceLimitsT = ResourceLimits>
  HyperParameterTuningJobConfig& WithResourceLimits(ResourceLimitsT&& value) {
    SetResourceLimits(std::forward<ResourceLimitsT>(value));
    return *this;
  }
 
 
  inline const ParameterRanges& GetParameterRanges() const { return m_parameterRanges; }
  inline bool ParameterRangesHasBeenSet() const { return m_parameterRangesHasBeenSet; }
  template <typename ParameterRangesT = ParameterRanges>
  void SetParameterRanges(ParameterRangesT&& value) {
    m_parameterRangesHasBeenSet = true;
    m_parameterRanges = std::forward<ParameterRangesT>(value);
  }
  template <typename ParameterRangesT = ParameterRanges>
  HyperParameterTuningJobConfig& WithParameterRanges(ParameterRangesT&& value) {
    SetParameterRanges(std::forward<ParameterRangesT>(value));
    return *this;
  }
 
 
  inline TrainingJobEarlyStoppingType GetTrainingJobEarlyStoppingType() const { return m_trainingJobEarlyStoppingType; }
  inline bool TrainingJobEarlyStoppingTypeHasBeenSet() const { return m_trainingJobEarlyStoppingTypeHasBeenSet; }
  inline void SetTrainingJobEarlyStoppingType(TrainingJobEarlyStoppingType value) {
    m_trainingJobEarlyStoppingTypeHasBeenSet = true;
    m_trainingJobEarlyStoppingType = value;
  }
  inline HyperParameterTuningJobConfig& WithTrainingJobEarlyStoppingType(TrainingJobEarlyStoppingType value) {
    SetTrainingJobEarlyStoppingType(value);
    return *this;
  }
 
 
  inline const TuningJobCompletionCriteria& GetTuningJobCompletionCriteria() const { return m_tuningJobCompletionCriteria; }
  inline bool TuningJobCompletionCriteriaHasBeenSet() const { return m_tuningJobCompletionCriteriaHasBeenSet; }
  template <typename TuningJobCompletionCriteriaT = TuningJobCompletionCriteria>
  void SetTuningJobCompletionCriteria(TuningJobCompletionCriteriaT&& value) {
    m_tuningJobCompletionCriteriaHasBeenSet = true;
    m_tuningJobCompletionCriteria = std::forward<TuningJobCompletionCriteriaT>(value);
  }
  template <typename TuningJobCompletionCriteriaT = TuningJobCompletionCriteria>
  HyperParameterTuningJobConfig& WithTuningJobCompletionCriteria(TuningJobCompletionCriteriaT&& value) {
    SetTuningJobCompletionCriteria(std::forward<TuningJobCompletionCriteriaT>(value));
    return *this;
  }
 
 
  inline int GetRandomSeed() const { return m_randomSeed; }
  inline bool RandomSeedHasBeenSet() const { return m_randomSeedHasBeenSet; }
  inline void SetRandomSeed(int value) {
    m_randomSeedHasBeenSet = true;
    m_randomSeed = value;
  }
  inline HyperParameterTuningJobConfig& WithRandomSeed(int value) {
    SetRandomSeed(value);
    return *this;
  }
 private:
  HyperParameterTuningJobStrategyType m_strategy{HyperParameterTuningJobStrategyType::NOT_SET};
 
  HyperParameterTuningJobStrategyConfig m_strategyConfig;
 
  HyperParameterTuningJobObjective m_hyperParameterTuningJobObjective;
 
  ResourceLimits m_resourceLimits;
 
  ParameterRanges m_parameterRanges;
 
  TrainingJobEarlyStoppingType m_trainingJobEarlyStoppingType{TrainingJobEarlyStoppingType::NOT_SET};
 
  TuningJobCompletionCriteria m_tuningJobCompletionCriteria;
 
  int m_randomSeed{0};
  bool m_strategyHasBeenSet = false;
  bool m_strategyConfigHasBeenSet = false;
  bool m_hyperParameterTuningJobObjectiveHasBeenSet = false;
  bool m_resourceLimitsHasBeenSet = false;
  bool m_parameterRangesHasBeenSet = false;
  bool m_trainingJobEarlyStoppingTypeHasBeenSet = false;
  bool m_tuningJobCompletionCriteriaHasBeenSet = false;
  bool m_randomSeedHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws