object LogLoss extends ClassificationLoss
Class for log loss calculation (for classification). This uses twice the binomial negative log likelihood, called "deviance" in Friedman (1999).
The log loss is defined as: 2 log(1 + exp(-2 y F(x))) where y is a label in {-1, 1} and F(x) is the model prediction for features x.
- Annotations
- @Since( "1.2.0" )
- Source
- LogLoss.scala
- Alphabetic
- By Inheritance
- LogLoss
- ClassificationLoss
- 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 loss gradients for the gradient boosting calculation for binary classification The gradient with respect to F(x) is: - 4 y / (1 + exp(2 y F(x)))