Class: SVMKit::EvaluationMeasure::Recall
- Inherits:
-
Object
- Object
- SVMKit::EvaluationMeasure::Recall
- Includes:
- Base::Evaluator
- Defined in:
- lib/svmkit/evaluation_measure/recall.rb
Overview
Recall is a class that calculates the recall of the predicted labels.
Instance Attribute Summary collapse
-
#average ⇒ String
readonly
Return the average type for calculation of recall.
Instance Method Summary collapse
-
#initialize(average: 'binary') ⇒ Recall
constructor
Create a new evaluation measure calculater for recall score.
-
#score(y_true, y_pred) ⇒ Float
Claculate average recall.
Constructor Details
#initialize(average: 'binary') ⇒ Recall
Create a new evaluation measure calculater for recall score.
25 26 27 |
# File 'lib/svmkit/evaluation_measure/recall.rb', line 25 def initialize(average: 'binary') @average = average end |
Instance Attribute Details
#average ⇒ String (readonly)
Return the average type for calculation of recall.
20 21 22 |
# File 'lib/svmkit/evaluation_measure/recall.rb', line 20 def average @average end |
Instance Method Details
#score(y_true, y_pred) ⇒ Float
Claculate average recall
34 35 36 37 38 39 40 41 42 43 |
# File 'lib/svmkit/evaluation_measure/recall.rb', line 34 def score(y_true, y_pred) case @average when 'binary' recall_each_class(y_true, y_pred).last when 'micro' micro_average_recall(y_true, y_pred) when 'macro' macro_average_recall(y_true, y_pred) end end |