RFTHyperParameters.h

 
#pragma once
#include <aws/bedrock/Bedrock_EXPORTS.h>
#include <aws/bedrock/model/ReasoningEffort.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace Bedrock {
namespace Model {
 
class RFTHyperParameters {
 public:
  AWS_BEDROCK_API RFTHyperParameters() = default;
  AWS_BEDROCK_API RFTHyperParameters(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCK_API RFTHyperParameters& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_BEDROCK_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline int GetEpochCount() const { return m_epochCount; }
  inline bool EpochCountHasBeenSet() const { return m_epochCountHasBeenSet; }
  inline void SetEpochCount(int value) {
    m_epochCountHasBeenSet = true;
    m_epochCount = value;
  }
  inline RFTHyperParameters& WithEpochCount(int value) {
    SetEpochCount(value);
    return *this;
  }
 
 
  inline int GetBatchSize() const { return m_batchSize; }
  inline bool BatchSizeHasBeenSet() const { return m_batchSizeHasBeenSet; }
  inline void SetBatchSize(int value) {
    m_batchSizeHasBeenSet = true;
    m_batchSize = value;
  }
  inline RFTHyperParameters& WithBatchSize(int value) {
    SetBatchSize(value);
    return *this;
  }
 
 
  inline double GetLearningRate() const { return m_learningRate; }
  inline bool LearningRateHasBeenSet() const { return m_learningRateHasBeenSet; }
  inline void SetLearningRate(double value) {
    m_learningRateHasBeenSet = true;
    m_learningRate = value;
  }
  inline RFTHyperParameters& WithLearningRate(double value) {
    SetLearningRate(value);
    return *this;
  }
 
 
  inline int GetMaxPromptLength() const { return m_maxPromptLength; }
  inline bool MaxPromptLengthHasBeenSet() const { return m_maxPromptLengthHasBeenSet; }
  inline void SetMaxPromptLength(int value) {
    m_maxPromptLengthHasBeenSet = true;
    m_maxPromptLength = value;
  }
  inline RFTHyperParameters& WithMaxPromptLength(int value) {
    SetMaxPromptLength(value);
    return *this;
  }
 
 
  inline int GetTrainingSamplePerPrompt() const { return m_trainingSamplePerPrompt; }
  inline bool TrainingSamplePerPromptHasBeenSet() const { return m_trainingSamplePerPromptHasBeenSet; }
  inline void SetTrainingSamplePerPrompt(int value) {
    m_trainingSamplePerPromptHasBeenSet = true;
    m_trainingSamplePerPrompt = value;
  }
  inline RFTHyperParameters& WithTrainingSamplePerPrompt(int value) {
    SetTrainingSamplePerPrompt(value);
    return *this;
  }
 
 
  inline int GetInferenceMaxTokens() const { return m_inferenceMaxTokens; }
  inline bool InferenceMaxTokensHasBeenSet() const { return m_inferenceMaxTokensHasBeenSet; }
  inline void SetInferenceMaxTokens(int value) {
    m_inferenceMaxTokensHasBeenSet = true;
    m_inferenceMaxTokens = value;
  }
  inline RFTHyperParameters& WithInferenceMaxTokens(int value) {
    SetInferenceMaxTokens(value);
    return *this;
  }
 
 
  inline ReasoningEffort GetReasoningEffort() const { return m_reasoningEffort; }
  inline bool ReasoningEffortHasBeenSet() const { return m_reasoningEffortHasBeenSet; }
  inline void SetReasoningEffort(ReasoningEffort value) {
    m_reasoningEffortHasBeenSet = true;
    m_reasoningEffort = value;
  }
  inline RFTHyperParameters& WithReasoningEffort(ReasoningEffort value) {
    SetReasoningEffort(value);
    return *this;
  }
 
 
  inline int GetEvalInterval() const { return m_evalInterval; }
  inline bool EvalIntervalHasBeenSet() const { return m_evalIntervalHasBeenSet; }
  inline void SetEvalInterval(int value) {
    m_evalIntervalHasBeenSet = true;
    m_evalInterval = value;
  }
  inline RFTHyperParameters& WithEvalInterval(int value) {
    SetEvalInterval(value);
    return *this;
  }
 private:
  int m_epochCount{0};
 
  int m_batchSize{0};
 
  double m_learningRate{0.0};
 
  int m_maxPromptLength{0};
 
  int m_trainingSamplePerPrompt{0};
 
  int m_inferenceMaxTokens{0};
 
  ReasoningEffort m_reasoningEffort{ReasoningEffort::NOT_SET};
 
  int m_evalInterval{0};
  bool m_epochCountHasBeenSet = false;
  bool m_batchSizeHasBeenSet = false;
  bool m_learningRateHasBeenSet = false;
  bool m_maxPromptLengthHasBeenSet = false;
  bool m_trainingSamplePerPromptHasBeenSet = false;
  bool m_inferenceMaxTokensHasBeenSet = false;
  bool m_reasoningEffortHasBeenSet = false;
  bool m_evalIntervalHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace Bedrock
}  // namespace Aws