Class: Spark::Mllib::LassoWithSGD

Inherits:
RegressionMethodBase show all
Defined in:
lib/spark/mllib/regression/lasso.rb

Constant Summary collapse

DEFAULT_OPTIONS = 
{
  iterations: 100,
  step: 1.0,
  reg_param: 0.01,
  mini_batch_fraction: 1.0,
  initial_weights: nil
}

Class Method Summary collapse

Class Method Details

.train(rdd, options = {}) ⇒ Object

Train a Lasso regression model on the given data.

Parameters:

rdd

The training data (RDD instance).

iterations

The number of iterations (default: 100).

step

The step parameter used in SGD (default: 1.0).

reg_param

The regularizer parameter (default: 0.0).

mini_batch_fraction

Fraction of data to be used for each SGD iteration (default: 1.0).

initial_weights

The initial weights (default: nil).



85
86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/spark/mllib/regression/lasso.rb', line 85

def self.train(rdd, options={})
  super

  weights, intercept = Spark.jb.call(RubyMLLibAPI.new, 'trainLassoModelWithSGD', rdd,
                                     options[:iterations].to_i,
                                     options[:step].to_f,
                                     options[:reg_param].to_f,
                                     options[:mini_batch_fraction].to_f,
                                     options[:initial_weights])

  LassoModel.new(weights, intercept)
end