Package | Description |
---|---|
org.apache.flink.ml.optimization |
Modifier and Type | Class and Description |
---|---|
class |
GradientDescent
Base class which performs Stochastic Gradient Descent optimization using mini batches.
|
class |
GradientDescentL1
Implementation of a SGD solver with L1 regularization.
|
class |
GradientDescentL2
Implementation of a SGD solver with L2 regularization.
|
class |
SimpleGradientDescent
Implementation of a SGD solver without regularization.
|
Modifier and Type | Method and Description |
---|---|
IterativeSolver |
IterativeSolver.setConvergenceThreshold(double convergenceThreshold) |
IterativeSolver |
IterativeSolver.setIterations(int iterations) |
IterativeSolver |
IterativeSolver.setLearningRateMethod(LearningRateMethod.LearningRateMethodTrait learningRateMethod) |
IterativeSolver |
IterativeSolver.setStepsize(double stepsize) |
Copyright © 2014–2017 The Apache Software Foundation. All rights reserved.