public class LinearRegression extends Object implements Serializable
Constructor and Description |
---|
LinearRegression() |
Modifier and Type | Method and Description |
---|---|
org.apache.spark.api.java.JavaRDD<scala.Tuple2<Double,Double>> |
test(org.apache.spark.mllib.regression.LinearRegressionModel linearRegressionModel,
org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> testingDataset)
This method applies linear regression using a given model and a dataset
|
org.apache.spark.mllib.regression.LinearRegressionModel |
train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingDataset,
int noOfIterations)
Linear regression train - overload method with 1 parameter
|
org.apache.spark.mllib.regression.LinearRegressionModel |
train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingDataset,
int noOfIterations,
double initialLearningRate,
double miniBatchFraction)
This method uses stochastic gradient descent (SGD) algorithm to train a linear regression model
|
public org.apache.spark.mllib.regression.LinearRegressionModel train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingDataset, int noOfIterations, double initialLearningRate, double miniBatchFraction)
trainingDataset
- Training dataset as a JavaRDD of LabeledPointsnoOfIterations
- Number of iterarationsinitialLearningRate
- Initial learning rate (SGD step size)miniBatchFraction
- SGD minibatch fractionpublic org.apache.spark.mllib.regression.LinearRegressionModel train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingDataset, int noOfIterations)
trainingDataset
- Training dataset as a JavaRDD of LabeledPointsnoOfIterations
- Number of iterarationspublic org.apache.spark.api.java.JavaRDD<scala.Tuple2<Double,Double>> test(org.apache.spark.mllib.regression.LinearRegressionModel linearRegressionModel, org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> testingDataset)
linearRegressionModel
- Linear regression modeltestingDataset
- Testing dataset as a JavaRDD of LabeledPointsCopyright © 2016 WSO2, Inc.. All Rights Reserved.