org.deeplearning4j.optimize.solvers

Class BackTrackLineSearch

java.lang.Object

org.deeplearning4j.optimize.solvers.BackTrackLineSearch

All Implemented Interfaces:

LineOptimizer

public class BackTrackLineSearch
extends Object
implements LineOptimizer

Constructor Summary

Constructors

Constructor and Description
BackTrackLineSearch(Model optimizable, BaseOptimizer optimizer)
BackTrackLineSearch(Model function, StepFunction stepFunction, BaseOptimizer optimizer)

Method Summary

Modifier and Type	Method and Description
double	getStpmax()
double	optimize(org.nd4j.linalg.api.ndarray.INDArray line, int lineSearchIteration, double initialStep, org.nd4j.linalg.api.ndarray.INDArray x, org.nd4j.linalg.api.ndarray.INDArray g) Returns the last step size used.
void	setAbsTolx(double tolx) Sets the tolerance of absolute diff in function value.
void	setRelTolx(double tolx) Sets the tolerance of relative diff in function value.
void	setStpmax(double stpmax)

Methods inherited from class java.lang.Object

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

Constructor Detail

BackTrackLineSearch

public BackTrackLineSearch(Model function,
                           StepFunction stepFunction,
                           BaseOptimizer optimizer)

BackTrackLineSearch

public BackTrackLineSearch(Model optimizable,
                           BaseOptimizer optimizer)

Method Detail

setStpmax

public void setStpmax(double stpmax)

getStpmax

public double getStpmax()

setRelTolx

public void setRelTolx(double tolx)

Sets the tolerance of relative diff in function value. Line search converges if abs(delta x / x) < tolx for all coordinates.

setAbsTolx

public void setAbsTolx(double tolx)

Sets the tolerance of absolute diff in function value. Line search converges if abs(delta x) < tolx for all coordinates.

optimize

public double optimize(org.nd4j.linalg.api.ndarray.INDArray line,
                       int lineSearchIteration,
                       double initialStep,
                       org.nd4j.linalg.api.ndarray.INDArray x,
                       org.nd4j.linalg.api.ndarray.INDArray g)
                throws InvalidStepException

Description copied from interface: LineOptimizer

Returns the last step size used.

Specified by:

optimize in interface LineOptimizer

Throws:

InvalidStepException