Class: Trainers::EvalPrediction

Inherits:

Object

• Object
• Trainers::EvalPrediction

Defined in:

lib/trainers/trainer_utils.rb

Instance Attribute Summary

• #label_ids ⇒ Object readonly

  Returns the value of attribute label_ids.

• #predictions ⇒ Object readonly

  Returns the value of attribute predictions.

Instance Method Summary

• #initialize(predictions:, label_ids:) ⇒ EvalPrediction constructor

  A new instance of EvalPrediction.

Constructor Details

#initialize(predictions:, label_ids:) ⇒ EvalPrediction

Returns a new instance of EvalPrediction.

# File 'lib/trainers/trainer_utils.rb', line 25

def initialize(predictions:, label_ids:)
  @predictions = predictions
  @label_ids   = label_ids
end

Instance Attribute Details

#label_ids ⇒ Object (readonly)

Returns the value of attribute label_ids.

# File 'lib/trainers/trainer_utils.rb', line 23

def label_ids
  @label_ids
end

#predictions ⇒ Object (readonly)

Returns the value of attribute predictions.

# File 'lib/trainers/trainer_utils.rb', line 23

def predictions
  @predictions
end