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 |
IterativeSolver
An abstract class for iterative optimization algorithms
|
class |
SimpleGradientDescent
Implementation of a SGD solver without regularization.
|
Modifier and Type | Method and Description |
---|---|
Solver |
Solver.setLossFunction(LossFunction lossFunction) |
Solver |
Solver.setRegularizationConstant(double regularizationConstant) |
Copyright © 2014–2017 The Apache Software Foundation. All rights reserved.