Class: Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RewardModelTrainingConfig

Inherits:
Object
  • Object
show all
Includes:
Core::Hashable, Core::JsonObjectSupport
Defined in:
lib/google/apis/aiplatform_v1beta1/classes.rb,
lib/google/apis/aiplatform_v1beta1/representations.rb,
lib/google/apis/aiplatform_v1beta1/representations.rb

Overview

Configures Reinforcement Learning to learn preference by training a reward model.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(**args) ⇒ GoogleCloudAiplatformV1beta1RewardModelTrainingConfig

Returns a new instance of GoogleCloudAiplatformV1beta1RewardModelTrainingConfig.



23919
23920
23921
 
# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 23919

def initialize(**args)
   update!(**args)
end

Instance Attribute Details

#epoch_count ⇒ Fixnum

Optional. Number of training epoches for the reward model training job. Corresponds to the JSON property epochCount

Returns:

  • (Fixnum) 


23912
23913
23914
 
# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 23912

def epoch_count
  @epoch_count
end

#learning_rate_multiplier ⇒ Float

Optional. Learning rate multiplier for reward model training. Corresponds to the JSON property learningRateMultiplier

Returns:

  • (Float) 


23917
23918
23919
 
# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 23917

def learning_rate_multiplier
  @learning_rate_multiplier
end

Instance Method Details

#update!(**args) ⇒ Object

Update properties of this object



23924
23925
23926
23927
 
# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 23924

def update!(**args)
  @epoch_count = args[:epoch_count] if args.key?(:epoch_count)
  @learning_rate_multiplier = args[:learning_rate_multiplier] if args.key?(:learning_rate_multiplier)
end