public class SquaredL2Updater extends Updater
Constructor and Description |
---|
SquaredL2Updater() |
Modifier and Type | Method and Description |
---|---|
scala.Tuple2<Vector,Object> |
compute(Vector weightsOld,
Vector gradient,
double stepSize,
int iter,
double regParam)
Compute an updated value for weights given the gradient, stepSize, iteration number and
regularization parameter.
|
public scala.Tuple2<Vector,Object> compute(Vector weightsOld, Vector gradient, double stepSize, int iter, double regParam)
Updater
compute
in class Updater
weightsOld
- - Column matrix of size dx1 where d is the number of features.gradient
- - Column matrix of size dx1 where d is the number of features.stepSize
- - step size across iterationsiter
- - Iteration numberregParam
- - Regularization parameter