rumale-svm 0.3.0 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e2fa12724b67f4cc5be042e64a8d29284e2292f23a5b13031e4b2ee33397285a
-  data.tar.gz: 453907d44a6113bf65531ec39e9d6da219bd1c2d0822cc6c72a1188e1b1e0b5e
+  metadata.gz: f0000e7a25f4aaf36f645a7860d13bed3ca06ea7d99bc15469cd7c9ea0a57288
+  data.tar.gz: 0db6f53cabe1a84493d20bf9e0822057dca4ec92042bdefb6fda15f794aba9aa
 SHA512:
-  metadata.gz: 490675646f497293d10112605ab1d5dcbc1a69a21e67f2712b304b89d7e542e9894358e536dec4811e04ead3bb94fbd77c5273a7e3464cc883e9e50e8212af0b
-  data.tar.gz: 9ccac4d7d7cf56deac34f9c9c731e3b6c455df65ca7349ed0634c37106ea0e6fe42dec243459f966e87c3aadf72525211fa4c365925c59f6089024689968c9be
+  metadata.gz: af0433aacb57daa0b9bf81dfe961f71ef71751215749201c18c1c81aa592d55050e4ed6d65605809ff6c36153cbbeb45e200181675dec23e13f884b70185ccab
+  data.tar.gz: c97c4a38ccdfabd869e6f5196eae73f32371d7bd12ff8e9612ede2cbfe89bb3bc563505429590ef65d928ba84d76cc06cf9c5f9eee0e79b161cbfd470da80cc0

data/CHANGELOG.md

@@ -1,3 +1,12 @@
+# 0.5.1
+- Refator specs and config files.
+
+# 0.5.0
+- Add type declaration files.
+
+# 0.4.0
+- Add linear one-class support vector machine.
+
 # 0.3.0
 - Fix to raise error when calling prediction method before training model.
 - Fix some config files.

data/LICENSE.txt

@@ -1,4 +1,4 @@
-Copyright (c) 2019-2020 Atsushi Tatsuma
+Copyright (c) 2019-2022 Atsushi Tatsuma
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without

data/README.md

@@ -1,10 +1,10 @@
 # Rumale::SVM
 
-[![Build Status](https://travis-ci.org/yoshoku/rumale-svm.svg?branch=master)](https://travis-ci.org/yoshoku/rumale-svm)
-[![Coverage Status](https://coveralls.io/repos/github/yoshoku/rumale-svm/badge.svg?branch=master)](https://coveralls.io/github/yoshoku/rumale-svm?branch=master)
+[![Build Status](https://github.com/yoshoku/rumale-svm/workflows/build/badge.svg)](https://github.com/yoshoku/rumale-svm/actions?query=workflow%3Abuild)
+[![Coverage Status](https://coveralls.io/repos/github/yoshoku/rumale-svm/badge.svg?branch=main)](https://coveralls.io/github/yoshoku/rumale-svm?branch=main)
 [![Gem Version](https://badge.fury.io/rb/rumale-svm.svg)](https://badge.fury.io/rb/rumale-svm)
-[![BSD 3-Clause License](https://img.shields.io/badge/License-BSD%203--Clause-orange.svg)](https://github.com/yoshoku/rumale-svm/blob/master/LICENSE.txt)
-[![Documentation](http://img.shields.io/badge/api-reference-blue.svg)](https://yoshoku.github.io/rumale-svm/doc/)
+[![BSD 3-Clause License](https://img.shields.io/badge/License-BSD%203--Clause-orange.svg)](https://github.com/yoshoku/rumale-svm/blob/main/LICENSE.txt)
+[![Documentation](https://img.shields.io/badge/api-reference-blue.svg)](https://yoshoku.github.io/rumale-svm/doc/)
 
 Rumale::SVM provides support vector machine algorithms in
 [LIBSVM](https://www.csie.ntu.edu.tw/~cjlin/libsvm/) and [LIBLINEAR](https://www.csie.ntu.edu.tw/~cjlin/liblinear/)
@@ -77,20 +77,10 @@ $ ruby rumale_svm_test.rb
 Accuracy: 0.835
 ```
 
-## Development
-
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
-
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
-
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/yoshoku/rumale-svm. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
+Bug reports and pull requests are welcome on GitHub at https://github.com/yoshoku/rumale-svm. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](https://contributor-covenant.org) code of conduct.
 
 ## License
 
 The gem is available as open source under the terms of the [BSD-3-Clause License](https://opensource.org/licenses/BSD-3-Clause).
-
-## Code of Conduct
-
-Everyone interacting in the Rumale::Svm project’s codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/yoshoku/rumale-svm/blob/master/CODE_OF_CONDUCT.md).

data/lib/rumale/svm/linear_one_class_svm.rb

@@ -0,0 +1,118 @@
+# frozen_string_literal: true
+
+require 'numo/liblinear'
+require 'rumale/base/base_estimator'
+require 'rumale/validation'
+
+module Rumale
+  module SVM
+    # LinearOneClassSVM is a class that provides linear One-class Support Vector Machine in LIBLINEAR with Rumale interface.
+    #
+    # @example
+    #   estimator = Rumale::SVM::LinearOneClassSVM.new(nu: 0.05, random_seed: 1)
+    #   estimator.fit(training_samples, traininig_labels)
+    #   results = estimator.predict(testing_samples)
+    class LinearOneClassSVM
+      include Base::BaseEstimator
+      include Validation
+
+      # Return the weight vector for LinearOneClassSVM.
+      # @return [Numo::DFloat] (shape: [n_features])
+      attr_reader :weight_vec
+
+      # Return the bias term (a.k.a. intercept) for LinearOneClassSVM.
+      # @return [Float]
+      attr_reader :bias_term
+
+      # Create a new estimator with linear One-class Support Vector Machine.
+      #
+      # @param nu [Float] The fraction of data as outliers. The interval of nu is (0, 1].
+      # @param reg_param [Float] The regularization parameter.
+      # @param tol [Float] The tolerance of termination criterion.
+      # @param verbose [Boolean] The flag indicating whether to output learning process message
+      # @param random_seed [Integer/Nil] The seed value using to initialize the random generator.
+      def initialize(nu: 0.05, reg_param: 1.0, tol: 1e-3, verbose: false, random_seed: nil)
+        check_params_numeric(nu: nu, reg_param: reg_param, tol: tol)
+        check_params_boolean(verbose: verbose)
+        check_params_numeric_or_nil(random_seed: random_seed)
+        @params = {}
+        @params[:nu] = nu.to_f
+        @params[:reg_param] = reg_param.to_f
+        @params[:tol] = tol.to_f
+        @params[:verbose] = verbose
+        @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
+      end
+
+      # Fit the model with given training data.
+      #
+      # @overload fit(x) -> LinearOneClassSVM
+      #   @param x [Numo::DFloat] (shape: [n_samples, n_features]) The training data to be used for fitting the model.
+      #
+      # @return [LinearOneClassSVM] The learned estimator itself.
+      def fit(x, _y = nil)
+        x = check_convert_sample_array(x)
+        dummy = Numo::DFloat.ones(x.shape[0])
+        @model = Numo::Liblinear.train(x, dummy, liblinear_params)
+        @weight_vec = @model[:w].dup
+        @bias_term = @model[:rho]
+        self
+      end
+
+      # Calculate confidence scores for samples.
+      #
+      # @param x [Numo::DFloat] (shape: [n_samples, n_features]) The samples to compute the scores.
+      # @return [Numo::DFloat] (shape: [n_samples, n_classes]) Confidence score per sample.
+      def decision_function(x)
+        raise "#{self.class.name}\##{__method__} expects to be called after training the model with the fit method." unless trained?
+        x = check_convert_sample_array(x)
+        Numo::Liblinear.decision_function(x, liblinear_params, @model)
+      end
+
+      # Predict class labels for samples.
+      #
+      # @param x [Numo::DFloat] (shape: [n_samples, n_features]) The samples to predict the labels.
+      # @return [Numo::Int32] (shape: [n_samples]) Predicted label per sample.
+      def predict(x)
+        raise "#{self.class.name}\##{__method__} expects to be called after training the model with the fit method." unless trained?
+        x = check_convert_sample_array(x)
+        Numo::Int32.cast(Numo::Liblinear.predict(x, liblinear_params, @model))
+      end
+
+      # Dump marshal data.
+      # @return [Hash] The marshal data about LinearOneClassSVM.
+      def marshal_dump
+        { params: @params,
+          model: @model,
+          weight_vec: @weight_vec,
+          bias_term: @bias_term }
+      end
+
+      # Load marshal data.
+      # @return [nil]
+      def marshal_load(obj)
+        @params = obj[:params]
+        @model = obj[:model]
+        @weight_vec = obj[:weight_vec]
+        @bias_term = obj[:bias_term]
+        nil
+      end
+
+      private
+
+      def liblinear_params
+        res = {}
+        res[:solver_type] = Numo::Liblinear::SolverType::ONECLASS_SVM
+        res[:eps] = @params[:tol]
+        res[:C] = @params[:reg_param]
+        res[:nu] = @params[:nu]
+        res[:verbose] = @params[:verbose]
+        res[:random_seed] = @params[:random_seed]
+        res
+      end
+
+      def trained?
+        !@model.nil?
+      end
+    end
+  end
+end

data/lib/rumale/svm/linear_svc.rb

@@ -58,10 +58,6 @@ module Rumale
         @params[:tol] = tol.to_f
         @params[:verbose] = verbose
         @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
-        @model = nil
-        @weight_vec = nil
-        @bias_term = nil
-        @prob_param = nil
       end
 
       # Fit the model with given training data.

data/lib/rumale/svm/linear_svr.rb

@@ -54,9 +54,6 @@ module Rumale
         @params[:tol] = tol.to_f
         @params[:verbose] = verbose
         @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
-        @model = nil
-        @weight_vec = nil
-        @bias_term = nil
       end
 
       # Fit the model with given training data.

data/lib/rumale/svm/logistic_regression.rb

@@ -52,9 +52,6 @@ module Rumale
         @params[:tol] = tol.to_f
         @params[:verbose] = verbose
         @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
-        @model = nil
-        @weight_vec = nil
-        @bias_term = nil
       end
 
       # Fit the model with given training data.

data/lib/rumale/svm/nu_svc.rb

@@ -47,7 +47,6 @@ module Rumale
         @params[:tol] = tol.to_f
         @params[:verbose] = verbose
         @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
-        @model = nil
       end
 
       # Fit the model with given training data.

data/lib/rumale/svm/nu_svr.rb

@@ -45,7 +45,6 @@ module Rumale
         @params[:tol] = tol.to_f
         @params[:verbose] = verbose
         @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
-        @model = nil
       end
 
       # Fit the model with given training data.

data/lib/rumale/svm/one_class_svm.rb

@@ -45,7 +45,6 @@ module Rumale
         @params[:tol] = tol.to_f
         @params[:verbose] = verbose
         @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
-        @model = nil
       end
 
       # Fit the model with given training data.

data/lib/rumale/svm/svc.rb

@@ -47,7 +47,6 @@ module Rumale
         @params[:tol] = tol.to_f
         @params[:verbose] = verbose
         @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
-        @model = nil
       end
 
       # Fit the model with given training data.

data/lib/rumale/svm/svr.rb

@@ -48,7 +48,6 @@ module Rumale
         @params[:tol] = tol.to_f
         @params[:verbose] = verbose
         @params[:random_seed] = random_seed.nil? ? nil : random_seed.to_i
-        @model = nil
       end
 
       # Fit the model with given training data.

data/lib/rumale/svm/version.rb

@@ -5,6 +5,6 @@ module Rumale
   # This module consists of Rumale interfaces for suppor vector machine algorithms with LIBSVM and LIBLINEAR.
   module SVM
     # The version of Rumale-SVM you are using.
-    VERSION = '0.3.0'
+    VERSION = '0.5.1'
   end
 end

data/lib/rumale/svm.rb

@@ -9,3 +9,4 @@ require 'rumale/svm/one_class_svm'
 require 'rumale/svm/linear_svc'
 require 'rumale/svm/linear_svr'
 require 'rumale/svm/logistic_regression'
+require 'rumale/svm/linear_one_class_svm'

data/sig/rumale/svm/linear_one_class_svm.rbs

@@ -0,0 +1,24 @@
+module Rumale
+  module SVM
+    class LinearOneClassSVM
+      include Base::BaseEstimator
+      include Validation
+
+      attr_reader weight_vec: Numo::DFloat
+      attr_reader bias_term: Float
+
+      def initialize: (?nu: Float nu, ?reg_param: Float reg_param, ?tol: Float tol,
+                       ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, ?untyped? _y) -> LinearOneClassSVM
+      def decision_function: (Numo::DFloat x) -> Numo::DFloat
+      def predict: (Numo::DFloat x) -> Numo::Int32
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped, weight_vec: Numo::DFloat, bias_term: Float }
+      def marshal_load: (untyped obj) -> void
+
+      private
+
+      def liblinear_params: () -> untyped
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm/linear_svc.rbs

@@ -0,0 +1,37 @@
+module Rumale
+  module SVM
+    class LinearSVC
+      include Base::BaseEstimator
+      include Base::Classifier
+
+      attr_reader weight_vec: Numo::DFloat
+      attr_reader bias_term: Numo::DFloat
+
+      def initialize: (?penalty: String penalty, ?loss: String loss, ?dual: bool dual, ?reg_param: Float reg_param,
+                       ?fit_bias: bool fit_bias, ?bias_scale: Float bias_scale, ?probability: bool probability,
+                       ?tol: Float tol, ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, Numo::Int32 y) -> LinearSVC
+      def decision_function: (Numo::DFloat x) -> Numo::DFloat
+      def predict: (Numo::DFloat x) -> Numo::Int32
+      def predict_proba: (Numo::DFloat x) -> Numo::DFloat
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped, weight_vec: Numo::DFloat, bias_term: Numo::DFloat, prob_param: Numo::DFloat }
+      def marshal_load: (Hash[Symbol, untyped] obj) -> void
+
+      private
+
+      def expand_feature: (Numo::DFloat x) -> Numo::DFloat
+      def weight_and_bias: (Numo::DFloat base_weight) -> [Numo::DFloat, Numo::DFloat]
+      def proba_model: (Numo::DFloat df, Numo::Int32 y) -> Numo::DFloat
+      def liblinear_params: () -> untyped
+      def solver_type: () -> Integer
+      def binary_class?: () -> bool
+      def fit_probability?: () -> bool
+      def fit_bias?: () -> bool
+      def bias_scale: () -> Float
+      def n_classes: () -> Integer
+      def n_features: () -> Integer
+      def labels: () -> Numo::Int32
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm/linear_svr.rbs

@@ -0,0 +1,29 @@
+module Rumale
+  module SVM
+    class LinearSVR
+      include Base::BaseEstimator
+      include Base::Regressor
+
+      attr_reader weight_vec: Numo::DFloat
+      attr_reader bias_term: Numo::DFloat
+
+      def initialize: (?loss: String loss, ?dual: bool dual, ?reg_param: Float reg_param, ?epsilon: Float epsilon,
+                       ?fit_bias: bool fit_bias, ?bias_scale: Float bias_scale, ?tol: Float tol,
+                       ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, Numo::DFloat y) -> LinearSVR
+      def predict: (Numo::DFloat x) -> Numo::DFloat
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped, weight_vec: Numo::DFloat, bias_term: Numo::DFloat }
+      def marshal_load: (Hash[Symbol, untyped] obj) -> void
+
+      private
+
+      def expand_feature: (Numo::DFloat x) -> Numo::DFloat
+      def weight_and_bias: (Numo::DFloat base_weight) -> [Numo::DFloat, Numo::DFloat]
+      def liblinear_params: () -> untyped
+      def solver_type: () -> Integer
+      def fit_bias?: () -> bool
+      def bias_scale: () -> Float
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm/logistic_regression.rbs

@@ -0,0 +1,34 @@
+module Rumale
+  module SVM
+    class LogisticRegression
+      include Base::BaseEstimator
+      include Base::Classifier
+
+      attr_reader weight_vec: Numo::DFloat
+      attr_reader bias_term: Numo::DFloat
+
+      def initialize: (?penalty: String penalty, ?dual: bool dual, ?reg_param: Float reg_param,
+                       ?fit_bias: bool fit_bias, ?bias_scale: Float bias_scale, ?tol: Float tol,
+                       ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, Numo::Int32 y) -> LogisticRegression
+      def decision_function: (Numo::DFloat x) -> Numo::DFloat
+      def predict: (Numo::DFloat x) -> Numo::Int32
+      def predict_proba: (Numo::DFloat x) -> Numo::DFloat
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped, weight_vec: Numo::DFloat, bias_term: Numo::DFloat }
+      def marshal_load: (untyped obj) -> void
+
+      private
+
+      def expand_feature: (Numo::DFloat x) -> Numo::DFloat
+      def weight_and_bias: (Numo::DFloat base_weight) -> [Numo::DFloat, Numo::DFloat]
+      def liblinear_params: () -> untyped
+      def solver_type: () -> Integer
+      def binary_class?: () -> bool
+      def fit_bias?: () -> bool
+      def bias_scale: () -> Float
+      def n_classes: () -> Integer
+      def n_features: () -> Integer
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm/nu_svc.rbs

@@ -0,0 +1,34 @@
+module Rumale
+  module SVM
+    class NuSVC
+      include Base::BaseEstimator
+      include Base::Classifier
+
+      def initialize: (?nu: Float nu, ?kernel: String kernel, ?degree: Integer degree, ?gamma: Float gamma, ?coef0: Float coef0,
+                       ?shrinking: bool shrinking, ?probability: bool probability,
+                       ?cache_size: Float cache_size, ?tol: Float tol,
+                       ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, Numo::Int32 y) -> NuSVC
+      def decision_function: (Numo::DFloat x) -> Numo::DFloat
+      def predict: (Numo::DFloat x) -> Numo::Int32
+      def predict_proba: (Numo::DFloat x) -> Numo::DFloat
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped }
+      def marshal_load: (untyped obj) -> void
+      def support: () -> Numo::Int32
+      def support_vectors: () -> Numo::DFloat
+      def n_support: () -> Numo::Int32
+      def duel_coef: () -> Numo::DFloat
+      def intercept: () -> Numo::DFloat
+      def prob_a: () -> Numo::DFloat
+      def prob_b: () -> Numo::DFloat
+
+      private
+
+      def add_index_col: (Numo::DFloat x) -> Numo::DFloat
+      def del_index_col: (Numo::DFloat x) -> Numo::DFloat
+      def precomputed_kernel?: () -> bool
+      def libsvm_params: () -> untyped
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm/nu_svr.rbs

@@ -0,0 +1,29 @@
+module Rumale
+  module SVM
+    class NuSVR
+      include Base::BaseEstimator
+      include Base::Regressor
+
+      def initialize: (?nu: Float nu, ?kernel: String kernel, ?degree: Integer degree, ?gamma: Float gamma, ?coef0: Float coef0,
+                       ?shrinking: bool shrinking, ?cache_size: Float cache_size, ?tol: Float tol,
+                       ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, Numo::DFloat y) -> NuSVR
+      def predict: (Numo::DFloat x) -> Numo::DFloat
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped }
+      def marshal_load: (Hash[Symbol, untyped] obj) -> void
+      def support: () -> Numo::Int32
+      def support_vectors: () -> Numo::DFloat
+      def n_support: () -> Integer
+      def duel_coef: () -> Numo::DFloat
+      def intercept: () -> Numo::DFloat
+
+      private
+
+      def add_index_col: (Numo::DFloat x) -> Numo::DFloat
+      def del_index_col: (Numo::DFloat x) -> Numo::DFloat
+      def precomputed_kernel?: () -> bool
+      def libsvm_params: () -> untyped
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm/one_class_svm.rbs

@@ -0,0 +1,27 @@
+module Rumale
+  module SVM
+    class OneClassSVM
+      include Base::BaseEstimator
+      include Validation
+
+      def initialize: (?nu: Float nu, ?kernel: String kernel, ?degree: Integer degree, ?gamma: Float gamma, ?coef0: Float coef0,
+                       ?shrinking: bool shrinking, ?cache_size: Float cache_size, ?tol: Float tol,
+                       ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, ?untyped? _y) -> OneClassSVM
+      def decision_function: (Numo::DFloat x) -> Numo::DFloat
+      def predict: (Numo::DFloat x) -> Numo::Int32
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped }
+      def marshal_load: (Hash[Symbol, untyped] obj) -> void
+      def support: () -> Numo::Int32
+      def support_vectors: () -> Numo::DFloat
+      def n_support: () -> Integer
+      def duel_coef: () -> Numo::DFloat
+      def intercept: () -> Numo::DFloat
+
+      private
+
+      def libsvm_params: () -> untyped
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm/svc.rbs

@@ -0,0 +1,34 @@
+module Rumale
+  module SVM
+    class SVC
+      include Base::BaseEstimator
+      include Base::Classifier
+
+      def initialize: (?reg_param: Float reg_param, ?kernel: String kernel, ?degree: Integer degree, ?gamma: Float gamma,
+                       ?coef0: Float coef0, ?shrinking: bool shrinking, ?probability: bool probability,
+                       ?cache_size: Float cache_size, ?tol: Float tol,
+                       ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, Numo::Int32 y) -> SVC
+      def decision_function: (Numo::DFloat x) -> Numo::DFloat
+      def predict: (Numo::DFloat x) -> Numo::Int32
+      def predict_proba: (Numo::DFloat x) -> Numo::DFloat
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped }
+      def marshal_load: (Hash[Symbol, untyped] obj) -> nil
+      def support: () -> Numo::Int32
+      def support_vectors: () -> Numo::DFloat
+      def n_support: () -> Numo::Int32
+      def duel_coef: () -> Numo::DFloat
+      def intercept: () -> Numo::DFloat
+      def prob_a: () -> Numo::DFloat
+      def prob_b: () -> Numo::DFloat
+
+      private
+
+      def add_index_col: (Numo::DFloat x) -> Numo::DFloat
+      def del_index_col: (Numo::DFloat x) -> Numo::DFloat
+      def precomputed_kernel?: () -> bool
+      def libsvm_params: () -> untyped
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm/svr.rbs

@@ -0,0 +1,30 @@
+module Rumale
+  module SVM
+    class SVR
+      include Base::BaseEstimator
+      include Base::Regressor
+
+      def initialize: (?reg_param: Float reg_param, ?epsilon: Float epsilon, ?kernel: String kernel, ?degree: Integer degree,
+                       ?gamma: Float gamma, ?coef0: Float coef0, ?shrinking: bool shrinking,
+                       ?cache_size: Float cache_size, ?tol: Float tol,
+                       ?verbose: bool verbose, ?random_seed: untyped? random_seed) -> void
+      def fit: (Numo::DFloat x, Numo::DFloat y) -> SVR
+      def predict: (Numo::DFloat x) -> Numo::DFloat
+      def marshal_dump: () -> { params: Hash[Symbol, untyped], model: untyped }
+      def marshal_load: (Hash[Symbol, untyped] obj) -> void
+      def support: () -> Numo::Int32
+      def support_vectors: () -> Numo::DFloat
+      def n_support: () -> Integer
+      def duel_coef: () -> Numo::DFloat
+      def intercept: () -> Numo::DFloat
+
+      private
+
+      def add_index_col: (Numo::DFloat x) -> Numo::DFloat
+      def del_index_col: (Numo::DFloat x) -> Numo::DFloat
+      def precomputed_kernel?: () -> bool
+      def libsvm_params: () -> untyped
+      def trained?: () -> bool
+    end
+  end
+end

data/sig/rumale/svm.rbs

@@ -0,0 +1,5 @@
+module Rumale
+  module SVM
+    VERSION: String
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rumale-svm
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.5.1
 platform: ruby
 authors:
 - yoshoku
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-06-06 00:00:00.000000000 Z
+date: 2022-05-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-liblinear
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '1.1'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.0'
+        version: '1.1'
 - !ruby/object:Gem::Dependency
   name: numo-libsvm
   requirement: !ruby/object:Gem::Requirement
@@ -45,6 +45,9 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.14'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '0.24'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -52,6 +55,9 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.14'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '0.24'
 description: 'Rumale-SVM provides support vector machine algorithms of LIBSVM and
   LIBLINEAR with Rumale interface.
 
@@ -62,21 +68,11 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".coveralls.yml"
-- ".github/workflows/build.yml"
-- ".gitignore"
-- ".rspec"
-- ".rubocop.yml"
-- ".travis.yml"
 - CHANGELOG.md
-- CODE_OF_CONDUCT.md
-- Gemfile
 - LICENSE.txt
 - README.md
-- Rakefile
-- bin/console
-- bin/setup
 - lib/rumale/svm.rb
+- lib/rumale/svm/linear_one_class_svm.rb
 - lib/rumale/svm/linear_svc.rb
 - lib/rumale/svm/linear_svr.rb
 - lib/rumale/svm/logistic_regression.rb
@@ -86,15 +82,25 @@ files:
 - lib/rumale/svm/svc.rb
 - lib/rumale/svm/svr.rb
 - lib/rumale/svm/version.rb
-- rumale-svm.gemspec
+- sig/rumale/svm.rbs
+- sig/rumale/svm/linear_one_class_svm.rbs
+- sig/rumale/svm/linear_svc.rbs
+- sig/rumale/svm/linear_svr.rbs
+- sig/rumale/svm/logistic_regression.rbs
+- sig/rumale/svm/nu_svc.rbs
+- sig/rumale/svm/nu_svr.rbs
+- sig/rumale/svm/one_class_svm.rbs
+- sig/rumale/svm/svc.rbs
+- sig/rumale/svm/svr.rbs
 homepage: https://github.com/yoshoku/rumale-svm
 licenses:
 - BSD-3-Clause
 metadata:
   homepage_uri: https://github.com/yoshoku/rumale-svm
   source_code_uri: https://github.com/yoshoku/rumale-svm
-  changelog_uri: https://github.com/yoshoku/rumale-svm/blob/master/CHANGELOG.md
+  changelog_uri: https://github.com/yoshoku/rumale-svm/blob/main/CHANGELOG.md
   documentation_uri: https://yoshoku.github.io/rumale-svm/doc/
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths:
@@ -110,7 +116,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
+rubygems_version: 3.2.33
 signing_key:
 specification_version: 4
 summary: Rumale-SVM provides support vector machine algorithms of LIBSVM and LIBLINEAR

data/.coveralls.yml

@@ -1 +0,0 @@
-service_name: travis-ci

data/.github/workflows/build.yml

@@ -1,24 +0,0 @@
-name: build
-
-on: [push]
-
-jobs:
-  build:
-
-    runs-on: ubuntu-latest
-
-    strategy:
-      matrix:
-        ruby: ['2.4.x', '2.5.x', '2.6.x']
-
-    steps:
-    - uses: actions/checkout@master
-    - name: Set up Ruby
-      uses: actions/setup-ruby@master
-      with:
-        ruby-version: ${{ matrix.ruby }}
-    - name: Build and test with Rake
-      run: |
-        gem install bundler
-        bundle install --jobs 4 --retry 3
-        bundle exec rake

data/.gitignore

@@ -1,18 +0,0 @@
-/.bundle/
-/.yardoc
-/Gemfile.lock
-/_yardoc/
-/coverage/
-/doc/
-/pkg/
-/spec/reports/
-/tmp/
-
-# rspec failure tracking
-.rspec_status
-
-*.swp
-*.bundle
-tags
-.DS_Store
-.ruby-version

data/.rspec

@@ -1,3 +0,0 @@
---format documentation
---color
---require spec_helper

data/.rubocop.yml

@@ -1,107 +0,0 @@
-require:
-  - rubocop-performance
-  - rubocop-rspec
-
-AllCops:
-  TargetRubyVersion: 2.4
-  DisplayCopNames: true
-  DisplayStyleGuide: true
-  Exclude:
-    - 'bin/*'
-    - 'rumale-svm.gemspec'
-    - 'Rakefile'
-    - 'Gemfile'
-
-Layout/EmptyLineAfterGuardClause:
-  Enabled: false
-
-Layout/EmptyLinesAroundAttributeAccessor:
-  Enabled: true
-
-Layout/LineLength:
-  Max: 145
-  IgnoredPatterns: ['(\A|\s)#']
-
-Layout/SpaceAroundMethodCallOperator:
-  Enabled: true
-
-Lint/DeprecatedOpenSSLConstant:
-  Enabled: true
-
-Lint/RaiseException:
-  Enabled: true
-
-Lint/StructNewOverride:
-  Enabled: true
-
-Metrics/ModuleLength:
-  Max: 200
-
-Metrics/ClassLength:
-  Max: 200
-
-Metrics/MethodLength:
-  Max: 40
-
-Metrics/AbcSize:
-  Max: 60
-
-Metrics/CyclomaticComplexity:
-  Max: 16
-
-Metrics/PerceivedComplexity:
-  Max: 16
-
-Metrics/BlockLength:
-  Max: 40
-  Exclude:
-    - 'spec/**/*'
-
-Metrics/ParameterLists:
-  Max: 12
-
-Naming/MethodParameterName:
-  Enabled: false
-
-Naming/ConstantName:
-  Enabled: false
-
-Security/MarshalLoad:
-  Enabled: false
-
-Style/Documentation:
-  Enabled: false
-
-Style/ExponentialNotation:
-  Enabled: true
-
-Style/HashEachMethods:
-  Enabled: true
-
-Style/HashTransformKeys:
-  Enabled: true
-
-Style/HashTransformValues:
-  Enabled: true
-
-Style/SlicingWithRange:
-  Enabled: true
-
-Style/FormatStringToken:
-  Enabled: false
-
-Style/NumericLiterals:
-  Enabled: false
-
-RSpec/MultipleExpectations:
-  Enabled: false
-
-RSpec/ExampleLength:
-  Max: 40
-
-RSpec/InstanceVariable:
-  Enabled: false
-
-RSpec/LeakyConstantDeclaration:
-  Enabled: false
-

data/.travis.yml

@@ -1,14 +0,0 @@
----
-sudo: false
-os: linux
-dist: bionic
-language: ruby
-cache: bundler
-rvm:
-  - '2.4'
-  - '2.5'
-  - '2.6'
-  - '2.7'
-
-before_install:
-  - gem install bundler -v 2.0.2

data/CODE_OF_CONDUCT.md

@@ -1,74 +0,0 @@
-# Contributor Covenant Code of Conduct
-
-## Our Pledge
-
-In the interest of fostering an open and welcoming environment, we as
-contributors and maintainers pledge to making participation in our project and
-our community a harassment-free experience for everyone, regardless of age, body
-size, disability, ethnicity, gender identity and expression, level of experience,
-nationality, personal appearance, race, religion, or sexual identity and
-orientation.
-
-## Our Standards
-
-Examples of behavior that contributes to creating a positive environment
-include:
-
-* Using welcoming and inclusive language
-* Being respectful of differing viewpoints and experiences
-* Gracefully accepting constructive criticism
-* Focusing on what is best for the community
-* Showing empathy towards other community members
-
-Examples of unacceptable behavior by participants include:
-
-* The use of sexualized language or imagery and unwelcome sexual attention or
-advances
-* Trolling, insulting/derogatory comments, and personal or political attacks
-* Public or private harassment
-* Publishing others' private information, such as a physical or electronic
-  address, without explicit permission
-* Other conduct which could reasonably be considered inappropriate in a
-  professional setting
-
-## Our Responsibilities
-
-Project maintainers are responsible for clarifying the standards of acceptable
-behavior and are expected to take appropriate and fair corrective action in
-response to any instances of unacceptable behavior.
-
-Project maintainers have the right and responsibility to remove, edit, or
-reject comments, commits, code, wiki edits, issues, and other contributions
-that are not aligned to this Code of Conduct, or to ban temporarily or
-permanently any contributor for other behaviors that they deem inappropriate,
-threatening, offensive, or harmful.
-
-## Scope
-
-This Code of Conduct applies both within project spaces and in public spaces
-when an individual is representing the project or its community. Examples of
-representing a project or community include using an official project e-mail
-address, posting via an official social media account, or acting as an appointed
-representative at an online or offline event. Representation of a project may be
-further defined and clarified by project maintainers.
-
-## Enforcement
-
-Instances of abusive, harassing, or otherwise unacceptable behavior may be
-reported by contacting the project team at yoshoku@outlook.com. All
-complaints will be reviewed and investigated and will result in a response that
-is deemed necessary and appropriate to the circumstances. The project team is
-obligated to maintain confidentiality with regard to the reporter of an incident.
-Further details of specific enforcement policies may be posted separately.
-
-Project maintainers who do not follow or enforce the Code of Conduct in good
-faith may face temporary or permanent repercussions as determined by other
-members of the project's leadership.
-
-## Attribution
-
-This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 1.4,
-available at [http://contributor-covenant.org/version/1/4][version]
-
-[homepage]: http://contributor-covenant.org
-[version]: http://contributor-covenant.org/version/1/4/

data/Gemfile

@@ -1,8 +0,0 @@
-source 'https://rubygems.org'
-
-# Specify your gem's dependencies in rumale-svm.gemspec
-gemspec
-
-gem 'coveralls', '~> 0.8'
-gem 'rake', '~> 12.0'
-gem 'rspec', '~> 3.0'

data/Rakefile

@@ -1,6 +0,0 @@
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-
-RSpec::Core::RakeTask.new(:spec)
-
-task :default => :spec

data/bin/console

@@ -1,14 +0,0 @@
-#!/usr/bin/env ruby
-
-require "bundler/setup"
-require "rumale/svm"
-
-# You can add fixtures and/or initialization code here to make experimenting
-# with your gem easier. You can also use a different console, if you like.
-
-# (If you use this, don't forget to add pry to your Gemfile!)
-# require "pry"
-# Pry.start
-
-require "irb"
-IRB.start(__FILE__)

data/bin/setup

@@ -1,8 +0,0 @@
-#!/usr/bin/env bash
-set -euo pipefail
-IFS=$'\n\t'
-set -vx
-
-bundle install
-
-# Do any other automated setup that you need to do here

data/rumale-svm.gemspec

@@ -1,37 +0,0 @@
-lib = File.expand_path('lib', __dir__)
-$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
-require 'rumale/svm/version'
-
-Gem::Specification.new do |spec|
-  spec.name          = 'rumale-svm'
-  spec.version       = Rumale::SVM::VERSION
-  spec.authors       = ['yoshoku']
-  spec.email         = ['yoshoku@outlook.com']
-
-  spec.summary       = <<~MSG
-    Rumale-SVM provides support vector machine algorithms of LIBSVM and LIBLINEAR with Rumale interface.
-  MSG
-  spec.description = <<~MSG
-    Rumale-SVM provides support vector machine algorithms of LIBSVM and LIBLINEAR with Rumale interface.
-  MSG
-  spec.homepage = 'https://github.com/yoshoku/rumale-svm'
-  spec.license = 'BSD-3-Clause'
-
-  spec.metadata['homepage_uri'] = spec.homepage
-  spec.metadata['source_code_uri'] = 'https://github.com/yoshoku/rumale-svm'
-  spec.metadata['changelog_uri'] = 'https://github.com/yoshoku/rumale-svm/blob/master/CHANGELOG.md'
-  spec.metadata['documentation_uri'] = 'https://yoshoku.github.io/rumale-svm/doc/'
-
-  # Specify which files should be added to the gem when it is released.
-  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
-  spec.files = Dir.chdir(File.expand_path(__dir__)) do
-    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
-  end
-  spec.bindir        = 'exe'
-  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
-  spec.require_paths = ['lib']
-
-  spec.add_runtime_dependency 'numo-liblinear', '~> 1.0'
-  spec.add_runtime_dependency 'numo-libsvm', '~> 1.0'
-  spec.add_runtime_dependency 'rumale', '~> 0.14'
-end