EvaluationDatasetMetricConfig.h

 
#pragma once
#include <aws/bedrock/Bedrock_EXPORTS.h>
#include <aws/bedrock/model/EvaluationDataset.h>
#include <aws/bedrock/model/EvaluationTaskType.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace Bedrock {
namespace Model {
 
class EvaluationDatasetMetricConfig {
 public:
  AWS_BEDROCK_API EvaluationDatasetMetricConfig() = default;
  AWS_BEDROCK_API EvaluationDatasetMetricConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCK_API EvaluationDatasetMetricConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCK_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline EvaluationTaskType GetTaskType() const { return m_taskType; }
  inline bool TaskTypeHasBeenSet() const { return m_taskTypeHasBeenSet; }
  inline void SetTaskType(EvaluationTaskType value) {
    m_taskTypeHasBeenSet = true;
    m_taskType = value;
  }
  inline EvaluationDatasetMetricConfig& WithTaskType(EvaluationTaskType value) {
    SetTaskType(value);
    return *this;
  }
 
 
  inline const EvaluationDataset& GetDataset() const { return m_dataset; }
  inline bool DatasetHasBeenSet() const { return m_datasetHasBeenSet; }
  template <typename DatasetT = EvaluationDataset>
  void SetDataset(DatasetT&& value) {
    m_datasetHasBeenSet = true;
    m_dataset = std::forward<DatasetT>(value);
  }
  template <typename DatasetT = EvaluationDataset>
  EvaluationDatasetMetricConfig& WithDataset(DatasetT&& value) {
    SetDataset(std::forward<DatasetT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Aws::String>& GetMetricNames() const { return m_metricNames; }
  inline bool MetricNamesHasBeenSet() const { return m_metricNamesHasBeenSet; }
  template <typename MetricNamesT = Aws::Vector<Aws::String>>
  void SetMetricNames(MetricNamesT&& value) {
    m_metricNamesHasBeenSet = true;
    m_metricNames = std::forward<MetricNamesT>(value);
  }
  template <typename MetricNamesT = Aws::Vector<Aws::String>>
  EvaluationDatasetMetricConfig& WithMetricNames(MetricNamesT&& value) {
    SetMetricNames(std::forward<MetricNamesT>(value));
    return *this;
  }
  template <typename MetricNamesT = Aws::String>
  EvaluationDatasetMetricConfig& AddMetricNames(MetricNamesT&& value) {
    m_metricNamesHasBeenSet = true;
    m_metricNames.emplace_back(std::forward<MetricNamesT>(value));
    return *this;
  }
 private:
  EvaluationTaskType m_taskType{EvaluationTaskType::NOT_SET};
  bool m_taskTypeHasBeenSet = false;
 
  EvaluationDataset m_dataset;
  bool m_datasetHasBeenSet = false;
 
  Aws::Vector<Aws::String> m_metricNames;
  bool m_metricNamesHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace Bedrock
}  // namespace Aws