Class: Rumale::EvaluationMeasure::Precision
- Inherits:
-
Object
- Object
- Rumale::EvaluationMeasure::Precision
- Includes:
- Base::Evaluator
- Defined in:
- lib/rumale/evaluation_measure/precision.rb
Overview
Precision is a class that calculates the preicision of the predicted labels.
Instance Attribute Summary collapse
-
#average ⇒ String
readonly
Return the average type for calculation of precision.
Instance Method Summary collapse
-
#initialize(average: 'binary') ⇒ Precision
constructor
Create a new evaluation measure calculater for precision score.
-
#score(y_true, y_pred) ⇒ Float
Calculate average precision.
Constructor Details
#initialize(average: 'binary') ⇒ Precision
Create a new evaluation measure calculater for precision score.
25 26 27 28 |
# File 'lib/rumale/evaluation_measure/precision.rb', line 25 def initialize(average: 'binary') check_params_string(average: average) @average = average end |
Instance Attribute Details
#average ⇒ String (readonly)
Return the average type for calculation of precision.
20 21 22 |
# File 'lib/rumale/evaluation_measure/precision.rb', line 20 def average @average end |
Instance Method Details
#score(y_true, y_pred) ⇒ Float
Calculate average precision.
35 36 37 38 39 40 41 42 43 44 45 46 47 |
# File 'lib/rumale/evaluation_measure/precision.rb', line 35 def score(y_true, y_pred) check_label_array(y_true) check_label_array(y_pred) case @average when 'binary' precision_each_class(y_true, y_pred).last when 'micro' micro_average_precision(y_true, y_pred) when 'macro' macro_average_precision(y_true, y_pred) end end |