org.apache.spark.mllib.regression

Class GeneralizedLinearModel

java.lang.Object

org.apache.spark.mllib.regression.GeneralizedLinearModel

All Implemented Interfaces:

java.io.Serializable

Direct Known Subclasses:

LassoModel, LinearRegressionModel, LogisticRegressionModel, RidgeRegressionModel, SVMModel

public abstract class GeneralizedLinearModel
extends java.lang.Object
implements scala.Serializable

:: DeveloperApi :: GeneralizedLinearModel (GLM) represents a model trained using GeneralizedLinearAlgorithm. GLMs consist of a weight vector and an intercept.

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

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GeneralizedLinearModel(Vector weights, double intercept)

Method Summary

Methods

Modifier and Type	Method and Description
double	intercept()
RDD<java.lang.Object>	predict(RDD<Vector> testData) Predict values for the given data set using the model trained.
double	predict(Vector testData) Predict values for a single data point using the model trained.
protected abstract double	predictPoint(Vector dataMatrix, Vector weightMatrix, double intercept) Predict the result given a data point and the weights learned.
java.lang.String	toString() Print a summary of the model.
Vector	weights()

Methods inherited from class java.lang.Object

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

Constructor Detail

GeneralizedLinearModel

public GeneralizedLinearModel(Vector weights,
                      double intercept)

Method Detail

weights

public Vector weights()

intercept

public double intercept()

predictPoint

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

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

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)

predict

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

Predict values for the given data set using the model trained.

Parameters:

testData - RDD representing data points to be predicted

Returns:

RDD[Double] where each entry contains the corresponding prediction

predict

public double predict(Vector testData)

Predict values for a single data point using the model trained.

Parameters:

testData - array representing a single data point

Returns:

Double prediction from the trained model

toString

public java.lang.String toString()

Print a summary of the model.

Overrides:

toString in class java.lang.Object

Returns:

(undocumented)