org.apache.spark.mllib.regression
Train a RidgeRegression model given an RDD of (label, features) pairs.
Train a RidgeRegression model given an RDD of (label, features) pairs. We run a fixed number of iterations of gradient descent using a step size of 1.0. We use the entire data set to compute the true gradient in each iteration.
RDD of (label, array of features) pairs.
Number of iterations of gradient descent to run.
a RidgeRegressionModel which has the weights and offset from training.
Train a RidgeRegression model given an RDD of (label, features) pairs.
Train a RidgeRegression model given an RDD of (label, features) pairs. We run a fixed number of iterations of gradient descent using the specified step size. We use the entire data set to compute the true gradient in each iteration.
RDD of (label, array of features) pairs.
Number of iterations of gradient descent to run.
Step size to be used for each iteration of Gradient Descent.
Regularization parameter.
a RidgeRegressionModel which has the weights and offset from training.
Train a RidgeRegression model given an RDD of (label, features) pairs.
Train a RidgeRegression model given an RDD of (label, features) pairs. We run a fixed number
of iterations of gradient descent using the specified step size. Each iteration uses
miniBatchFraction
fraction of the data to calculate a stochastic gradient.
RDD of (label, array of features) pairs.
Number of iterations of gradient descent to run.
Step size to be used for each iteration of gradient descent.
Regularization parameter.
Fraction of data to be used per iteration.
Train a RidgeRegression model given an RDD of (label, features) pairs.
Train a RidgeRegression model given an RDD of (label, features) pairs. We run a fixed number
of iterations of gradient descent using the specified step size. Each iteration uses
miniBatchFraction
fraction of the data to calculate a stochastic gradient. The weights used
in gradient descent are initialized using the initial weights provided.
RDD of (label, array of features) pairs.
Number of iterations of gradient descent to run.
Step size to be used for each iteration of gradient descent.
Regularization parameter.
Fraction of data to be used per iteration.
Initial set of weights to be used. Array should be equal in size to the number of features in the data.
Top-level methods for calling RidgeRegression.
(Since version 2.0.0)