TabularJobConfig.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/AutoMLJobCompletionCriteria.h>
#include <aws/sagemaker/model/AutoMLMode.h>
#include <aws/sagemaker/model/CandidateGenerationConfig.h>
#include <aws/sagemaker/model/ProblemType.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class TabularJobConfig {
 public:
  AWS_SAGEMAKER_API TabularJobConfig() = default;
  AWS_SAGEMAKER_API TabularJobConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API TabularJobConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const CandidateGenerationConfig& GetCandidateGenerationConfig() const { return m_candidateGenerationConfig; }
  inline bool CandidateGenerationConfigHasBeenSet() const { return m_candidateGenerationConfigHasBeenSet; }
  template <typename CandidateGenerationConfigT = CandidateGenerationConfig>
  void SetCandidateGenerationConfig(CandidateGenerationConfigT&& value) {
    m_candidateGenerationConfigHasBeenSet = true;
    m_candidateGenerationConfig = std::forward<CandidateGenerationConfigT>(value);
  }
  template <typename CandidateGenerationConfigT = CandidateGenerationConfig>
  TabularJobConfig& WithCandidateGenerationConfig(CandidateGenerationConfigT&& value) {
    SetCandidateGenerationConfig(std::forward<CandidateGenerationConfigT>(value));
    return *this;
  }
 
 
  inline const AutoMLJobCompletionCriteria& GetCompletionCriteria() const { return m_completionCriteria; }
  inline bool CompletionCriteriaHasBeenSet() const { return m_completionCriteriaHasBeenSet; }
  template <typename CompletionCriteriaT = AutoMLJobCompletionCriteria>
  void SetCompletionCriteria(CompletionCriteriaT&& value) {
    m_completionCriteriaHasBeenSet = true;
    m_completionCriteria = std::forward<CompletionCriteriaT>(value);
  }
  template <typename CompletionCriteriaT = AutoMLJobCompletionCriteria>
  TabularJobConfig& WithCompletionCriteria(CompletionCriteriaT&& value) {
    SetCompletionCriteria(std::forward<CompletionCriteriaT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetFeatureSpecificationS3Uri() const { return m_featureSpecificationS3Uri; }
  inline bool FeatureSpecificationS3UriHasBeenSet() const { return m_featureSpecificationS3UriHasBeenSet; }
  template <typename FeatureSpecificationS3UriT = Aws::String>
  void SetFeatureSpecificationS3Uri(FeatureSpecificationS3UriT&& value) {
    m_featureSpecificationS3UriHasBeenSet = true;
    m_featureSpecificationS3Uri = std::forward<FeatureSpecificationS3UriT>(value);
  }
  template <typename FeatureSpecificationS3UriT = Aws::String>
  TabularJobConfig& WithFeatureSpecificationS3Uri(FeatureSpecificationS3UriT&& value) {
    SetFeatureSpecificationS3Uri(std::forward<FeatureSpecificationS3UriT>(value));
    return *this;
  }
 
 
  inline AutoMLMode GetMode() const { return m_mode; }
  inline bool ModeHasBeenSet() const { return m_modeHasBeenSet; }
  inline void SetMode(AutoMLMode value) {
    m_modeHasBeenSet = true;
    m_mode = value;
  }
  inline TabularJobConfig& WithMode(AutoMLMode value) {
    SetMode(value);
    return *this;
  }
 
 
  inline bool GetGenerateCandidateDefinitionsOnly() const { return m_generateCandidateDefinitionsOnly; }
  inline bool GenerateCandidateDefinitionsOnlyHasBeenSet() const { return m_generateCandidateDefinitionsOnlyHasBeenSet; }
  inline void SetGenerateCandidateDefinitionsOnly(bool value) {
    m_generateCandidateDefinitionsOnlyHasBeenSet = true;
    m_generateCandidateDefinitionsOnly = value;
  }
  inline TabularJobConfig& WithGenerateCandidateDefinitionsOnly(bool value) {
    SetGenerateCandidateDefinitionsOnly(value);
    return *this;
  }
 
 
  inline ProblemType GetProblemType() const { return m_problemType; }
  inline bool ProblemTypeHasBeenSet() const { return m_problemTypeHasBeenSet; }
  inline void SetProblemType(ProblemType value) {
    m_problemTypeHasBeenSet = true;
    m_problemType = value;
  }
  inline TabularJobConfig& WithProblemType(ProblemType value) {
    SetProblemType(value);
    return *this;
  }
 
 
  inline const Aws::String& GetTargetAttributeName() const { return m_targetAttributeName; }
  inline bool TargetAttributeNameHasBeenSet() const { return m_targetAttributeNameHasBeenSet; }
  template <typename TargetAttributeNameT = Aws::String>
  void SetTargetAttributeName(TargetAttributeNameT&& value) {
    m_targetAttributeNameHasBeenSet = true;
    m_targetAttributeName = std::forward<TargetAttributeNameT>(value);
  }
  template <typename TargetAttributeNameT = Aws::String>
  TabularJobConfig& WithTargetAttributeName(TargetAttributeNameT&& value) {
    SetTargetAttributeName(std::forward<TargetAttributeNameT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetSampleWeightAttributeName() const { return m_sampleWeightAttributeName; }
  inline bool SampleWeightAttributeNameHasBeenSet() const { return m_sampleWeightAttributeNameHasBeenSet; }
  template <typename SampleWeightAttributeNameT = Aws::String>
  void SetSampleWeightAttributeName(SampleWeightAttributeNameT&& value) {
    m_sampleWeightAttributeNameHasBeenSet = true;
    m_sampleWeightAttributeName = std::forward<SampleWeightAttributeNameT>(value);
  }
  template <typename SampleWeightAttributeNameT = Aws::String>
  TabularJobConfig& WithSampleWeightAttributeName(SampleWeightAttributeNameT&& value) {
    SetSampleWeightAttributeName(std::forward<SampleWeightAttributeNameT>(value));
    return *this;
  }
 private:
  CandidateGenerationConfig m_candidateGenerationConfig;
 
  AutoMLJobCompletionCriteria m_completionCriteria;
 
  Aws::String m_featureSpecificationS3Uri;
 
  AutoMLMode m_mode{AutoMLMode::NOT_SET};
 
  bool m_generateCandidateDefinitionsOnly{false};
 
  ProblemType m_problemType{ProblemType::NOT_SET};
 
  Aws::String m_targetAttributeName;
 
  Aws::String m_sampleWeightAttributeName;
  bool m_candidateGenerationConfigHasBeenSet = false;
  bool m_completionCriteriaHasBeenSet = false;
  bool m_featureSpecificationS3UriHasBeenSet = false;
  bool m_modeHasBeenSet = false;
  bool m_generateCandidateDefinitionsOnlyHasBeenSet = false;
  bool m_problemTypeHasBeenSet = false;
  bool m_targetAttributeNameHasBeenSet = false;
  bool m_sampleWeightAttributeNameHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws