org.apache.spark.mllib.tree.loss

Class AbsoluteError

Object

org.apache.spark.mllib.tree.loss.AbsoluteError

All Implemented Interfaces:

java.io.Serializable, Loss

public class AbsoluteError
extends Object
implements Loss

:: DeveloperApi :: Class for absolute error loss calculation (for regression).

The absolute (L1) error is defined as: |y - F(x)| where y is the label and F(x) is the model prediction for features x.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
AbsoluteError()

Method Summary

Methods

Modifier and Type	Method and Description
static double	computeError(TreeEnsembleModel model, RDD<LabeledPoint> data) Method to calculate loss of the base learner for the gradient boosting calculation.
static double	gradient(TreeEnsembleModel model, LabeledPoint point) Method to calculate the gradients for the gradient boosting calculation for least absolute error calculation.

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.spark.mllib.tree.loss.Loss

computeError, gradient

Constructor Detail

AbsoluteError

public AbsoluteError()

Method Detail

gradient

public static double gradient(TreeEnsembleModel model,
              LabeledPoint point)

Method to calculate the gradients for the gradient boosting calculation for least absolute error calculation. The gradient with respect to F(x) is: sign(F(x) - y)

Parameters:

model - Ensemble model

point - Instance of the training dataset

Returns:

Loss gradient

computeError

public static double computeError(TreeEnsembleModel model,
                  RDD<LabeledPoint> data)

Method to calculate loss of the base learner for the gradient boosting calculation. Note: This method is not used by the gradient boosting algorithm but is useful for debugging purposes.

Parameters:

model - Ensemble model

data - Training dataset: RDD of LabeledPoint.

Returns:

Mean absolute error of model on data