TextGenerationJobConfig.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/AutoMLJobCompletionCriteria.h>
#include <aws/sagemaker/model/ModelAccessConfig.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class TextGenerationJobConfig {
 public:
  AWS_SAGEMAKER_API TextGenerationJobConfig() = default;
  AWS_SAGEMAKER_API TextGenerationJobConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API TextGenerationJobConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  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>
  TextGenerationJobConfig& WithCompletionCriteria(CompletionCriteriaT&& value) {
    SetCompletionCriteria(std::forward<CompletionCriteriaT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetBaseModelName() const { return m_baseModelName; }
  inline bool BaseModelNameHasBeenSet() const { return m_baseModelNameHasBeenSet; }
  template <typename BaseModelNameT = Aws::String>
  void SetBaseModelName(BaseModelNameT&& value) {
    m_baseModelNameHasBeenSet = true;
    m_baseModelName = std::forward<BaseModelNameT>(value);
  }
  template <typename BaseModelNameT = Aws::String>
  TextGenerationJobConfig& WithBaseModelName(BaseModelNameT&& value) {
    SetBaseModelName(std::forward<BaseModelNameT>(value));
    return *this;
  }
 
 
  inline const Aws::Map<Aws::String, Aws::String>& GetTextGenerationHyperParameters() const { return m_textGenerationHyperParameters; }
  inline bool TextGenerationHyperParametersHasBeenSet() const { return m_textGenerationHyperParametersHasBeenSet; }
  template <typename TextGenerationHyperParametersT = Aws::Map<Aws::String, Aws::String>>
  void SetTextGenerationHyperParameters(TextGenerationHyperParametersT&& value) {
    m_textGenerationHyperParametersHasBeenSet = true;
    m_textGenerationHyperParameters = std::forward<TextGenerationHyperParametersT>(value);
  }
  template <typename TextGenerationHyperParametersT = Aws::Map<Aws::String, Aws::String>>
  TextGenerationJobConfig& WithTextGenerationHyperParameters(TextGenerationHyperParametersT&& value) {
    SetTextGenerationHyperParameters(std::forward<TextGenerationHyperParametersT>(value));
    return *this;
  }
  template <typename TextGenerationHyperParametersKeyT = Aws::String, typename TextGenerationHyperParametersValueT = Aws::String>
  TextGenerationJobConfig& AddTextGenerationHyperParameters(TextGenerationHyperParametersKeyT&& key,
                                                            TextGenerationHyperParametersValueT&& value) {
    m_textGenerationHyperParametersHasBeenSet = true;
    m_textGenerationHyperParameters.emplace(std::forward<TextGenerationHyperParametersKeyT>(key),
                                            std::forward<TextGenerationHyperParametersValueT>(value));
    return *this;
  }
 
 
  inline const ModelAccessConfig& GetModelAccessConfig() const { return m_modelAccessConfig; }
  inline bool ModelAccessConfigHasBeenSet() const { return m_modelAccessConfigHasBeenSet; }
  template <typename ModelAccessConfigT = ModelAccessConfig>
  void SetModelAccessConfig(ModelAccessConfigT&& value) {
    m_modelAccessConfigHasBeenSet = true;
    m_modelAccessConfig = std::forward<ModelAccessConfigT>(value);
  }
  template <typename ModelAccessConfigT = ModelAccessConfig>
  TextGenerationJobConfig& WithModelAccessConfig(ModelAccessConfigT&& value) {
    SetModelAccessConfig(std::forward<ModelAccessConfigT>(value));
    return *this;
  }
 private:
  AutoMLJobCompletionCriteria m_completionCriteria;
 
  Aws::String m_baseModelName;
 
  Aws::Map<Aws::String, Aws::String> m_textGenerationHyperParameters;
 
  ModelAccessConfig m_modelAccessConfig;
  bool m_completionCriteriaHasBeenSet = false;
  bool m_baseModelNameHasBeenSet = false;
  bool m_textGenerationHyperParametersHasBeenSet = false;
  bool m_modelAccessConfigHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws