ClarifyInferenceConfig.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/ClarifyFeatureType.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class ClarifyInferenceConfig {
 public:
  AWS_SAGEMAKER_API ClarifyInferenceConfig() = default;
  AWS_SAGEMAKER_API ClarifyInferenceConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API ClarifyInferenceConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const Aws::String& GetFeaturesAttribute() const { return m_featuresAttribute; }
  inline bool FeaturesAttributeHasBeenSet() const { return m_featuresAttributeHasBeenSet; }
  template <typename FeaturesAttributeT = Aws::String>
  void SetFeaturesAttribute(FeaturesAttributeT&& value) {
    m_featuresAttributeHasBeenSet = true;
    m_featuresAttribute = std::forward<FeaturesAttributeT>(value);
  }
  template <typename FeaturesAttributeT = Aws::String>
  ClarifyInferenceConfig& WithFeaturesAttribute(FeaturesAttributeT&& value) {
    SetFeaturesAttribute(std::forward<FeaturesAttributeT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetContentTemplate() const { return m_contentTemplate; }
  inline bool ContentTemplateHasBeenSet() const { return m_contentTemplateHasBeenSet; }
  template <typename ContentTemplateT = Aws::String>
  void SetContentTemplate(ContentTemplateT&& value) {
    m_contentTemplateHasBeenSet = true;
    m_contentTemplate = std::forward<ContentTemplateT>(value);
  }
  template <typename ContentTemplateT = Aws::String>
  ClarifyInferenceConfig& WithContentTemplate(ContentTemplateT&& value) {
    SetContentTemplate(std::forward<ContentTemplateT>(value));
    return *this;
  }
 
 
  inline int GetMaxRecordCount() const { return m_maxRecordCount; }
  inline bool MaxRecordCountHasBeenSet() const { return m_maxRecordCountHasBeenSet; }
  inline void SetMaxRecordCount(int value) {
    m_maxRecordCountHasBeenSet = true;
    m_maxRecordCount = value;
  }
  inline ClarifyInferenceConfig& WithMaxRecordCount(int value) {
    SetMaxRecordCount(value);
    return *this;
  }
 
 
  inline int GetMaxPayloadInMB() const { return m_maxPayloadInMB; }
  inline bool MaxPayloadInMBHasBeenSet() const { return m_maxPayloadInMBHasBeenSet; }
  inline void SetMaxPayloadInMB(int value) {
    m_maxPayloadInMBHasBeenSet = true;
    m_maxPayloadInMB = value;
  }
  inline ClarifyInferenceConfig& WithMaxPayloadInMB(int value) {
    SetMaxPayloadInMB(value);
    return *this;
  }
 
 
  inline int GetProbabilityIndex() const { return m_probabilityIndex; }
  inline bool ProbabilityIndexHasBeenSet() const { return m_probabilityIndexHasBeenSet; }
  inline void SetProbabilityIndex(int value) {
    m_probabilityIndexHasBeenSet = true;
    m_probabilityIndex = value;
  }
  inline ClarifyInferenceConfig& WithProbabilityIndex(int value) {
    SetProbabilityIndex(value);
    return *this;
  }
 
 
  inline int GetLabelIndex() const { return m_labelIndex; }
  inline bool LabelIndexHasBeenSet() const { return m_labelIndexHasBeenSet; }
  inline void SetLabelIndex(int value) {
    m_labelIndexHasBeenSet = true;
    m_labelIndex = value;
  }
  inline ClarifyInferenceConfig& WithLabelIndex(int value) {
    SetLabelIndex(value);
    return *this;
  }
 
 
  inline const Aws::String& GetProbabilityAttribute() const { return m_probabilityAttribute; }
  inline bool ProbabilityAttributeHasBeenSet() const { return m_probabilityAttributeHasBeenSet; }
  template <typename ProbabilityAttributeT = Aws::String>
  void SetProbabilityAttribute(ProbabilityAttributeT&& value) {
    m_probabilityAttributeHasBeenSet = true;
    m_probabilityAttribute = std::forward<ProbabilityAttributeT>(value);
  }
  template <typename ProbabilityAttributeT = Aws::String>
  ClarifyInferenceConfig& WithProbabilityAttribute(ProbabilityAttributeT&& value) {
    SetProbabilityAttribute(std::forward<ProbabilityAttributeT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetLabelAttribute() const { return m_labelAttribute; }
  inline bool LabelAttributeHasBeenSet() const { return m_labelAttributeHasBeenSet; }
  template <typename LabelAttributeT = Aws::String>
  void SetLabelAttribute(LabelAttributeT&& value) {
    m_labelAttributeHasBeenSet = true;
    m_labelAttribute = std::forward<LabelAttributeT>(value);
  }
  template <typename LabelAttributeT = Aws::String>
  ClarifyInferenceConfig& WithLabelAttribute(LabelAttributeT&& value) {
    SetLabelAttribute(std::forward<LabelAttributeT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Aws::String>& GetLabelHeaders() const { return m_labelHeaders; }
  inline bool LabelHeadersHasBeenSet() const { return m_labelHeadersHasBeenSet; }
  template <typename LabelHeadersT = Aws::Vector<Aws::String>>
  void SetLabelHeaders(LabelHeadersT&& value) {
    m_labelHeadersHasBeenSet = true;
    m_labelHeaders = std::forward<LabelHeadersT>(value);
  }
  template <typename LabelHeadersT = Aws::Vector<Aws::String>>
  ClarifyInferenceConfig& WithLabelHeaders(LabelHeadersT&& value) {
    SetLabelHeaders(std::forward<LabelHeadersT>(value));
    return *this;
  }
  template <typename LabelHeadersT = Aws::String>
  ClarifyInferenceConfig& AddLabelHeaders(LabelHeadersT&& value) {
    m_labelHeadersHasBeenSet = true;
    m_labelHeaders.emplace_back(std::forward<LabelHeadersT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Aws::String>& GetFeatureHeaders() const { return m_featureHeaders; }
  inline bool FeatureHeadersHasBeenSet() const { return m_featureHeadersHasBeenSet; }
  template <typename FeatureHeadersT = Aws::Vector<Aws::String>>
  void SetFeatureHeaders(FeatureHeadersT&& value) {
    m_featureHeadersHasBeenSet = true;
    m_featureHeaders = std::forward<FeatureHeadersT>(value);
  }
  template <typename FeatureHeadersT = Aws::Vector<Aws::String>>
  ClarifyInferenceConfig& WithFeatureHeaders(FeatureHeadersT&& value) {
    SetFeatureHeaders(std::forward<FeatureHeadersT>(value));
    return *this;
  }
  template <typename FeatureHeadersT = Aws::String>
  ClarifyInferenceConfig& AddFeatureHeaders(FeatureHeadersT&& value) {
    m_featureHeadersHasBeenSet = true;
    m_featureHeaders.emplace_back(std::forward<FeatureHeadersT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<ClarifyFeatureType>& GetFeatureTypes() const { return m_featureTypes; }
  inline bool FeatureTypesHasBeenSet() const { return m_featureTypesHasBeenSet; }
  template <typename FeatureTypesT = Aws::Vector<ClarifyFeatureType>>
  void SetFeatureTypes(FeatureTypesT&& value) {
    m_featureTypesHasBeenSet = true;
    m_featureTypes = std::forward<FeatureTypesT>(value);
  }
  template <typename FeatureTypesT = Aws::Vector<ClarifyFeatureType>>
  ClarifyInferenceConfig& WithFeatureTypes(FeatureTypesT&& value) {
    SetFeatureTypes(std::forward<FeatureTypesT>(value));
    return *this;
  }
  inline ClarifyInferenceConfig& AddFeatureTypes(ClarifyFeatureType value) {
    m_featureTypesHasBeenSet = true;
    m_featureTypes.push_back(value);
    return *this;
  }
 private:
  Aws::String m_featuresAttribute;
 
  Aws::String m_contentTemplate;
 
  int m_maxRecordCount{0};
 
  int m_maxPayloadInMB{0};
 
  int m_probabilityIndex{0};
 
  int m_labelIndex{0};
 
  Aws::String m_probabilityAttribute;
 
  Aws::String m_labelAttribute;
 
  Aws::Vector<Aws::String> m_labelHeaders;
 
  Aws::Vector<Aws::String> m_featureHeaders;
 
  Aws::Vector<ClarifyFeatureType> m_featureTypes;
  bool m_featuresAttributeHasBeenSet = false;
  bool m_contentTemplateHasBeenSet = false;
  bool m_maxRecordCountHasBeenSet = false;
  bool m_maxPayloadInMBHasBeenSet = false;
  bool m_probabilityIndexHasBeenSet = false;
  bool m_labelIndexHasBeenSet = false;
  bool m_probabilityAttributeHasBeenSet = false;
  bool m_labelAttributeHasBeenSet = false;
  bool m_labelHeadersHasBeenSet = false;
  bool m_featureHeadersHasBeenSet = false;
  bool m_featureTypesHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws