Class: Rumale::LinearModel::Ridge
- Inherits:
-
BaseLinearModel
- Object
- BaseLinearModel
- Rumale::LinearModel::Ridge
- Includes:
- Base::Regressor
- Defined in:
- lib/rumale/linear_model/ridge.rb
Overview
Ridge is a class that implements Ridge Regression with mini-batch stochastic gradient descent optimization or singular value decomposition.
Instance Attribute Summary collapse
-
#bias_term ⇒ Numo::DFloat
readonly
Return the bias term (a.k.a. intercept).
-
#rng ⇒ Random
readonly
Return the random generator for random sampling.
-
#weight_vec ⇒ Numo::DFloat
readonly
Return the weight vector.
Attributes included from Base::BaseEstimator
Instance Method Summary collapse
-
#fit(x, y) ⇒ Ridge
Fit the model with given training data.
-
#initialize(reg_param: 1.0, fit_bias: false, bias_scale: 1.0, max_iter: 1000, batch_size: 10, optimizer: nil, solver: 'sgd', n_jobs: nil, random_seed: nil) ⇒ Ridge
constructor
Create a new Ridge regressor.
-
#marshal_dump ⇒ Hash
Dump marshal data.
-
#marshal_load(obj) ⇒ nil
Load marshal data.
-
#predict(x) ⇒ Numo::DFloat
Predict values for samples.
Methods included from Base::Regressor
Constructor Details
#initialize(reg_param: 1.0, fit_bias: false, bias_scale: 1.0, max_iter: 1000, batch_size: 10, optimizer: nil, solver: 'sgd', n_jobs: nil, random_seed: nil) ⇒ Ridge
Create a new Ridge regressor.
57 58 59 60 61 62 63 64 65 66 67 68 69 |
# File 'lib/rumale/linear_model/ridge.rb', line 57 def initialize(reg_param: 1.0, fit_bias: false, bias_scale: 1.0, max_iter: 1000, batch_size: 10, optimizer: nil, solver: 'sgd', n_jobs: nil, random_seed: nil) check_params_float(reg_param: reg_param, bias_scale: bias_scale) check_params_integer(max_iter: max_iter, batch_size: batch_size) check_params_boolean(fit_bias: fit_bias) check_params_string(solver: solver) check_params_type_or_nil(Integer, n_jobs: n_jobs, random_seed: random_seed) check_params_positive(reg_param: reg_param, max_iter: max_iter, batch_size: batch_size) keywd_args = method(:initialize).parameters.map { |_t, arg| [arg, binding.local_variable_get(arg)] }.to_h keywd_args.delete(:solver) super(keywd_args) @params[:solver] = solver != 'svd' ? 'sgd' : 'svd' end |
Instance Attribute Details
#bias_term ⇒ Numo::DFloat (readonly)
Return the bias term (a.k.a. intercept).
31 32 33 |
# File 'lib/rumale/linear_model/ridge.rb', line 31 def bias_term @bias_term end |
#rng ⇒ Random (readonly)
Return the random generator for random sampling.
35 36 37 |
# File 'lib/rumale/linear_model/ridge.rb', line 35 def rng @rng end |
#weight_vec ⇒ Numo::DFloat (readonly)
Return the weight vector.
27 28 29 |
# File 'lib/rumale/linear_model/ridge.rb', line 27 def weight_vec @weight_vec end |
Instance Method Details
#fit(x, y) ⇒ Ridge
Fit the model with given training data.
76 77 78 79 80 81 82 83 84 85 86 87 88 |
# File 'lib/rumale/linear_model/ridge.rb', line 76 def fit(x, y) check_sample_array(x) check_tvalue_array(y) check_sample_tvalue_size(x, y) if @params[:solver] == 'svd' && enable_linalg? fit_svd(x, y) else fit_sgd(x, y) end self end |
#marshal_dump ⇒ Hash
Dump marshal data.
101 102 103 104 105 106 |
# File 'lib/rumale/linear_model/ridge.rb', line 101 def marshal_dump { params: @params, weight_vec: @weight_vec, bias_term: @bias_term, rng: @rng } end |
#marshal_load(obj) ⇒ nil
Load marshal data.
110 111 112 113 114 115 116 |
# File 'lib/rumale/linear_model/ridge.rb', line 110 def marshal_load(obj) @params = obj[:params] @weight_vec = obj[:weight_vec] @bias_term = obj[:bias_term] @rng = obj[:rng] nil end |
#predict(x) ⇒ Numo::DFloat
Predict values for samples.
94 95 96 97 |
# File 'lib/rumale/linear_model/ridge.rb', line 94 def predict(x) check_sample_array(x) x.dot(@weight_vec.transpose) + @bias_term end |