org.apache.spark.mllib.tree.loss

Class AbsoluteError

Object

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

public class AbsoluteError
extends Object

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.

Constructor Summary

Constructors

Constructor and Description
AbsoluteError()

Method Summary

Modifier and Type	Method and Description
static double	gradient(double prediction, double label) 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

Constructor Detail

AbsoluteError

public AbsoluteError()

Method Detail

gradient

public static double gradient(double prediction,
                              double label)

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:

prediction - Predicted label.

label - True label.

Returns:

Loss gradient