org.apache.spark.mllib.regression

Class LinearRegressionModel

java.lang.Object

org.apache.spark.mllib.regression.GeneralizedLinearModel

org.apache.spark.mllib.regression.LinearRegressionModel

All Implemented Interfaces:

java.io.Serializable, PMMLExportable, RegressionModel, Saveable

public class LinearRegressionModel
extends GeneralizedLinearModel
implements RegressionModel, scala.Serializable, Saveable, PMMLExportable

Regression model trained using LinearRegression.

param: weights Weights computed for every feature. param: intercept Intercept computed for this model.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
LinearRegressionModel(Vector weights, double intercept)

Method Summary

Methods

Modifier and Type	Method and Description
protected java.lang.String	formatVersion() Current version of model save/load format.
double	intercept()
static LinearRegressionModel	load(SparkContext sc, java.lang.String path)
static JavaRDD<java.lang.Double>	predict(JavaRDD<Vector> testData)
static RDD<java.lang.Object>	predict(RDD<Vector> testData)
static double	predict(Vector testData)
protected double	predictPoint(Vector dataMatrix, Vector weightMatrix, double intercept) Predict the result given a data point and the weights learned.
void	save(SparkContext sc, java.lang.String path) Save this model to the given path.
static java.lang.String	toPMML()
static void	toPMML(java.io.OutputStream outputStream)
static void	toPMML(SparkContext sc, java.lang.String path)
static void	toPMML(java.lang.String localPath)
static java.lang.String	toString()
Vector	weights()

Methods inherited from class org.apache.spark.mllib.regression.GeneralizedLinearModel

predict, predict, toString

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.apache.spark.mllib.regression.RegressionModel

predict, predict, predict

Methods inherited from interface org.apache.spark.mllib.pmml.PMMLExportable

toPMML, toPMML, toPMML, toPMML, toPMML

Constructor Detail

LinearRegressionModel

public LinearRegressionModel(Vector weights,
                     double intercept)

Method Detail

load

public static LinearRegressionModel load(SparkContext sc,
                         java.lang.String path)

predict

public static RDD<java.lang.Object> predict(RDD<Vector> testData)

predict

public static double predict(Vector testData)

toString

public static java.lang.String toString()

predict

public static JavaRDD<java.lang.Double> predict(JavaRDD<Vector> testData)

toPMML

public static void toPMML(java.lang.String localPath)

toPMML

public static void toPMML(SparkContext sc,
          java.lang.String path)

toPMML

public static void toPMML(java.io.OutputStream outputStream)

toPMML

public static java.lang.String toPMML()

weights

public Vector weights()

Overrides:

weights in class GeneralizedLinearModel

intercept

public double intercept()

Overrides:

intercept in class GeneralizedLinearModel

predictPoint

protected double predictPoint(Vector dataMatrix,
                  Vector weightMatrix,
                  double intercept)

Description copied from class: GeneralizedLinearModel

Predict the result given a data point and the weights learned.

Specified by:

predictPoint in class GeneralizedLinearModel

Parameters:

dataMatrix - Row vector containing the features for this data point

weightMatrix - Column vector containing the weights of the model

intercept - Intercept of the model.

Returns:

(undocumented)

save

public void save(SparkContext sc,
        java.lang.String path)

Description copied from interface: Saveable

Save this model to the given path.

This saves: - human-readable (JSON) model metadata to path/metadata/ - Parquet formatted data to path/data/

The model may be loaded using Loader.load.

Specified by:

save in interface Saveable

Parameters:

sc - Spark context used to save model data.

path - Path specifying the directory in which to save this model. If the directory already exists, this method throws an exception.

formatVersion

protected java.lang.String formatVersion()

Description copied from interface: Saveable

Current version of model save/load format.

Specified by:

formatVersion in interface Saveable