Method: SVMKit::EvaluationMeasure::Accuracy#score

Defined in:
lib/svmkit/evaluation_measure/accuracy.rb

#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
24
25
26
# File 'lib/svmkit/evaluation_measure/accuracy.rb', line 21

def score(y_true, y_pred)
  SVMKit::Validation.check_label_array(y_true)
  SVMKit::Validation.check_label_array(y_pred)

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