Contents

# frozen_string_literal: true

require 'svmkit/validation'
require 'svmkit/evaluation_measure/r2_score'

module SVMKit
  module Base
    # Module for all regressors in SVMKit.
    module Regressor
      # An abstract method for fitting a model.
      def fit
        raise NotImplementedError, "#{__method__} has to be implemented in #{self.class}."
      end

      # An abstract method for predicting labels.
      def predict
        raise NotImplementedError, "#{__method__} has to be implemented in #{self.class}."
      end

      # Calculate the coefficient of determination for the given testing data.
      #
      # @param x [Numo::DFloat] (shape: [n_samples, n_features]) Testing data.
      # @param y [Numo::DFloat] (shape: [n_samples, n_outputs]) Target values for testing data.
      # @return [Float] Coefficient of determination
      def score(x, y)
        SVMKit::Validation.check_sample_array(x)
        SVMKit::Validation.check_tvalue_array(y)
        SVMKit::Validation.check_sample_tvalue_size(x, y)
        evaluator = SVMKit::EvaluationMeasure::R2Score.new
        evaluator.score(y, predict(x))
      end
    end
  end
end

Version data entries

17 entries across 17 versions & 1 rubygems

Version	Path
svmkit-0.7.3	lib/svmkit/base/regressor.rb
svmkit-0.7.2	lib/svmkit/base/regressor.rb
svmkit-0.7.1	lib/svmkit/base/regressor.rb
svmkit-0.7.0	lib/svmkit/base/regressor.rb
svmkit-0.6.3	lib/svmkit/base/regressor.rb
svmkit-0.6.2	lib/svmkit/base/regressor.rb
svmkit-0.6.1	lib/svmkit/base/regressor.rb
svmkit-0.6.0	lib/svmkit/base/regressor.rb
svmkit-0.5.2	lib/svmkit/base/regressor.rb
svmkit-0.5.1	lib/svmkit/base/regressor.rb
svmkit-0.5.0	lib/svmkit/base/regressor.rb
svmkit-0.4.1	lib/svmkit/base/regressor.rb
svmkit-0.4.0	lib/svmkit/base/regressor.rb
svmkit-0.3.3	lib/svmkit/base/regressor.rb
svmkit-0.3.2	lib/svmkit/base/regressor.rb
svmkit-0.3.1	lib/svmkit/base/regressor.rb
svmkit-0.3.0	lib/svmkit/base/regressor.rb