Sha256: 748e820e90810c28b310c3f5390a060c1845f300b3e3ce537d9c397d31a6d340

Contents?: true

Size: 1.13 KB

Versions: 8

Compression:

Stored size: 1.13 KB

Contents

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

Version data entries

8 entries across 8 versions & 1 rubygems

Version Path
rumale-svm-0.11.0 sig/rumale/svm/linear_svr.rbs
rumale-svm-0.10.0 sig/rumale/svm/linear_svr.rbs
rumale-svm-0.9.0 sig/rumale/svm/linear_svr.rbs
rumale-svm-0.8.0 sig/rumale/svm/linear_svr.rbs
rumale-svm-0.7.0 sig/rumale/svm/linear_svr.rbs
rumale-svm-0.6.0 sig/rumale/svm/linear_svr.rbs
rumale-svm-0.5.1 sig/rumale/svm/linear_svr.rbs
rumale-svm-0.5.0 sig/rumale/svm/linear_svr.rbs