abstract class Gradient extends Serializable
Class used to compute the gradient for a loss function, given a single data point.
- Source
- Gradient.scala
- Alphabetic
- By Inheritance
- Gradient
- Serializable
- Serializable
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Instance Constructors
- new Gradient()
Abstract Value Members
-
abstract
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
Concrete Value Members
-
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)