org.apache.spark.mllib.regression
Train a LinearRegression model given an RDD of (label, features) pairs.
Train a LinearRegression 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. Each pair describes a row of the data matrix A as well as the corresponding right hand side label y
Number of iterations of gradient descent to run.
a LinearRegressionModel which has the weights and offset from training.
Train a LinearRegression model given an RDD of (label, features) pairs.
Train a LinearRegression 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. Each pair describes a row of the data matrix A as well as the corresponding right hand side label y
Number of iterations of gradient descent to run.
Step size to be used for each iteration of Gradient Descent.
a LinearRegressionModel which has the weights and offset from training.
Train a LinearRegression model given an RDD of (label, features) pairs.
Train a LinearRegression 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. Each pair describes a row of the data matrix A as well as the corresponding right hand side label y
Number of iterations of gradient descent to run.
Step size to be used for each iteration of gradient descent.
Fraction of data to be used per iteration.
Train a Linear Regression model given an RDD of (label, features) pairs.
Train a Linear Regression 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. Each pair describes a row of the data matrix A as well as the corresponding right hand side label y
Number of iterations of gradient descent to run.
Step size to be used for each iteration of gradient descent.
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 LinearRegression.
(Since version 2.0.0) Use ml.regression.LinearRegression or LBFGS