Sha256: 4c9551c9ea56f9db798b275b2130d5e124ab5297ab0b92daf6f600a30dcdeae9
Contents?: true
Size: 1018 Bytes
Versions: 17
Compression:
Stored size: 1018 Bytes
Contents
# frozen_string_literal: true
require 'svmkit/validation'
require 'svmkit/base/evaluator'
module SVMKit
module EvaluationMeasure
# MeanAbsoluteError is a class that calculates the mean absolute error.
#
# @example
# evaluator = SVMKit::EvaluationMeasure::MeanAbsoluteError.new
# puts evaluator.score(ground_truth, predicted)
class MeanAbsoluteError
include Base::Evaluator
# Calculate mean absolute error.
#
# @param y_true [Numo::DFloat] (shape: [n_samples, n_outputs]) Ground truth target values.
# @param y_pred [Numo::DFloat] (shape: [n_samples, n_outputs]) Estimated target values.
# @return [Float] Mean absolute error
def score(y_true, y_pred)
SVMKit::Validation.check_tvalue_array(y_true)
SVMKit::Validation.check_tvalue_array(y_pred)
raise ArgumentError, 'Expect to have the same size both y_true and y_pred.' unless y_true.shape == y_pred.shape
(y_true - y_pred).abs.mean
end
end
end
end
Version data entries
17 entries across 17 versions & 1 rubygems