|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |
Object org.apache.spark.mllib.optimization.Gradient
public abstract class Gradient
:: DeveloperApi :: Class used to compute the gradient for a loss function, given a single data point.
Constructor Summary | |
---|---|
Gradient()
|
Method Summary | |
---|---|
scala.Tuple2<Vector,Object> |
compute(Vector data,
double label,
Vector weights)
Compute the gradient and loss given the features of a single data point. |
abstract double |
compute(Vector data,
double label,
Vector weights,
Vector cumGradient)
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. |
Methods inherited from class Object |
---|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
Constructor Detail |
---|
public Gradient()
Method Detail |
---|
public scala.Tuple2<Vector,Object> compute(Vector data, double label, Vector weights)
data
- features for one data pointlabel
- label for this data pointweights
- weights/coefficients corresponding to features
public abstract double compute(Vector data, double label, Vector weights, Vector cumGradient)
data
- features for one data pointlabel
- label for this data pointweights
- weights/coefficients corresponding to featurescumGradient
- the computed gradient will be added to this vector
|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |