Class: SVMKit::EvaluationMeasure::Accuracy

Inherits:
Object
  • Object
show all
Includes:
Base::Evaluator
Defined in:
lib/svmkit/evaluation_measure/accuracy.rb

Overview

Accuracy is a class that calculates the accuracy of classifier from the predicted labels.

Examples:

evaluator = SVMKit::EvaluationMeasure::Accuracy.new
puts evaluator.score(ground_truth, predicted)

Instance Method Summary collapse

Instance Method Details

#score(y_true, y_pred) ⇒ Float

Claculate mean accuracy.

Parameters:

  • y_true (Numo::Int32)

    (shape: [n_samples]) Ground truth labels.

  • y_pred (Numo::Int32)

    (shape: [n_samples]) Predicted labels.

Returns:

  • (Float)

    Mean accuracy



21
22
23
# File 'lib/svmkit/evaluation_measure/accuracy.rb', line 21

def score(y_true, y_pred)
  (y_true.to_a.map.with_index { |label, n| label == y_pred[n] ? 1 : 0 }).inject(:+) / y_true.size.to_f
end