Sha256: 582b1564aaee18a8fb9eb3d3256ff71e3870a86ed2dc12071db2b7b2ffdfd4d5
Contents?: true
Size: 849 Bytes
Versions: 8
Compression:
Stored size: 849 Bytes
Contents
# frozen_string_literal: true
require 'rumale/base/evaluator'
module Rumale
module EvaluationMeasure
# Accuracy is a class that calculates the accuracy of classifier from the predicted labels.
#
# @example
# require 'rumale/evaluation_measure/accuracy'
#
# evaluator = Rumale::EvaluationMeasure::Accuracy.new
# puts evaluator.score(ground_truth, predicted)
class Accuracy
include ::Rumale::Base::Evaluator
# Calculate mean accuracy.
#
# @param y_true [Numo::Int32] (shape: [n_samples]) Ground truth labels.
# @param y_pred [Numo::Int32] (shape: [n_samples]) Predicted labels.
# @return [Float] Mean accuracy
def score(y_true, y_pred)
(y_true.to_a.map.with_index { |label, n| label == y_pred[n] ? 1 : 0 }).sum / y_true.size.to_f
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems