object SquaredError extends Loss
Class for squared error loss calculation.
The squared (L2) error is defined as: (y - F(x))**2 where y is the label and F(x) is the model prediction for features x.
- Annotations
- @Since( "1.2.0" )
- Source
- SquaredError.scala
- Alphabetic
- By Inheritance
- SquaredError
- Loss
- Serializable
- Serializable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Value Members
-
def
computeError(model: TreeEnsembleModel, data: RDD[LabeledPoint]): Double
Method to calculate error of the base learner for the gradient boosting calculation.
Method to calculate error of the base learner for the gradient boosting calculation.
- model
Model of the weak learner.
- data
Training dataset: RDD of org.apache.spark.mllib.regression.LabeledPoint.
- returns
Measure of model error on data
- Definition Classes
- Loss
- Annotations
- @Since( "1.2.0" )
- Note
This method is not used by the gradient boosting algorithm but is useful for debugging purposes.
-
def
gradient(prediction: Double, label: Double): Double
Method to calculate the gradients for the gradient boosting calculation for least squares error calculation.
Method to calculate the gradients for the gradient boosting calculation for least squares error calculation. The gradient with respect to F(x) is: - 2 (y - F(x))
- prediction
Predicted label.
- label
True label.
- returns
Loss gradient
- Definition Classes
- SquaredError → Loss
- Annotations
- @Since( "1.2.0" )