org.apache.flink.ml.optimization

Class IterativeSolver

java.lang.Object

org.apache.flink.ml.optimization.Solver

org.apache.flink.ml.optimization.IterativeSolver

All Implemented Interfaces:

Serializable, WithParameters

Direct Known Subclasses:

GradientDescent

public abstract class IterativeSolver
extends Solver

An abstract class for iterative optimization algorithms

See Iterative Methods on Wikipedia for more info

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	IterativeSolver.ConvergenceThreshold$
static class	IterativeSolver.Iterations$
static class	IterativeSolver.LearningRate$
static class	IterativeSolver.LearningRateMethodValue$

Nested classes/interfaces inherited from class org.apache.flink.ml.optimization.Solver

Solver.LossFunction$, Solver.RegularizationConstant$

Constructor Summary

Constructors

Constructor and Description
IterativeSolver()

Method Summary

Modifier and Type	Method and Description
static double	MAX_DLOSS()
IterativeSolver	setConvergenceThreshold(double convergenceThreshold)
IterativeSolver	setIterations(int iterations)
IterativeSolver	setLearningRateMethod(LearningRateMethod.LearningRateMethodTrait learningRateMethod)
IterativeSolver	setStepsize(double stepsize)

Methods inherited from class org.apache.flink.ml.optimization.Solver

createInitialWeightsDS, createInitialWeightVector, optimize, setLossFunction, setRegularizationConstant

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.flink.ml.common.WithParameters

parameters

Constructor Detail

IterativeSolver

public IterativeSolver()

Method Detail

MAX_DLOSS

public static double MAX_DLOSS()

setIterations

public IterativeSolver setIterations(int iterations)

setStepsize

public IterativeSolver setStepsize(double stepsize)

setConvergenceThreshold

public IterativeSolver setConvergenceThreshold(double convergenceThreshold)

setLearningRateMethod

public IterativeSolver setLearningRateMethod(LearningRateMethod.LearningRateMethodTrait learningRateMethod)