class HingeGradient extends Gradient
Compute gradient and loss for a Hinge loss function, as used in SVM binary classification. See also the documentation for the precise formulation.
- Source
- Gradient.scala
- Note
This assumes that the labels are {0,1}
- Alphabetic
- By Inheritance
- HingeGradient
- Gradient
- Serializable
- Serializable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Instance Constructors
- new HingeGradient()
Value Members
-
def
compute(data: Vector, label: Double, weights: Vector, cumGradient: Vector): Double
Compute the gradient and loss given the features of a single data point, add the gradient to a provided vector to avoid creating new objects, and return loss.
Compute the gradient and loss given the features of a single data point, add the gradient to a provided vector to avoid creating new objects, and return loss.
- data
features for one data point
- label
label for this data point
- weights
weights/coefficients corresponding to features
- cumGradient
the computed gradient will be added to this vector
- returns
loss
- Definition Classes
- HingeGradient → Gradient
-
def
compute(data: Vector, label: Double, weights: Vector): (Vector, Double)
Compute the gradient and loss given the features of a single data point.
Compute the gradient and loss given the features of a single data point.
- data
features for one data point
- label
label for this data point
- weights
weights/coefficients corresponding to features
- returns
(gradient: Vector, loss: Double)
- Definition Classes
- HingeGradient → Gradient