org.deeplearning4j.optimize.optimizers

Class BackPropOptimizer

java.lang.Object

org.deeplearning4j.optimize.optimizers.BackPropOptimizer

All Implemented Interfaces:

Serializable, IterationListener, OptimizableByGradientValueMatrix

public class BackPropOptimizer
extends Object
implements Serializable, OptimizableByGradientValueMatrix, IterationListener

Optimizes via back prop gradients

Author:

Adam Gibson

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected NeuralNetPlotter	plotter

Constructor Summary

Constructors

Constructor and Description
BackPropOptimizer(BaseMultiLayerNetwork network, double lr, int epochs)

Method Summary

Modifier and Type	Method and Description
int	getNumParameters()
double	getParameter(int index)
org.nd4j.linalg.api.ndarray.INDArray	getParameters()
double	getValue()
org.nd4j.linalg.api.ndarray.INDArray	getValueGradient(int iteration)
void	iterationDone(int iteration) Event listener for each iteration
void	optimize(TrainingEvaluator eval)
void	setCurrentIteration(int value)
void	setParameter(int index, double value)
void	setParameters(org.nd4j.linalg.api.ndarray.INDArray params)

Methods inherited from class java.lang.Object

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

Field Detail

plotter

protected NeuralNetPlotter plotter

Constructor Detail

BackPropOptimizer

public BackPropOptimizer(BaseMultiLayerNetwork network,
                         double lr,
                         int epochs)

Method Detail

setCurrentIteration

public void setCurrentIteration(int value)

Specified by:

setCurrentIteration in interface OptimizableByGradientValueMatrix

optimize

public void optimize(TrainingEvaluator eval)

getValue

public double getValue()

Specified by:

getValue in interface OptimizableByGradientValueMatrix

getNumParameters

public int getNumParameters()

Specified by:

getNumParameters in interface OptimizableByGradientValueMatrix

setParameter

public void setParameter(int index,
                         double value)

Specified by:

setParameter in interface OptimizableByGradientValueMatrix

getParameters

public org.nd4j.linalg.api.ndarray.INDArray getParameters()

Specified by:

getParameters in interface OptimizableByGradientValueMatrix

getParameter

public double getParameter(int index)

Specified by:

getParameter in interface OptimizableByGradientValueMatrix

setParameters

public void setParameters(org.nd4j.linalg.api.ndarray.INDArray params)

Specified by:

setParameters in interface OptimizableByGradientValueMatrix

getValueGradient

public org.nd4j.linalg.api.ndarray.INDArray getValueGradient(int iteration)

Specified by:

getValueGradient in interface OptimizableByGradientValueMatrix

iterationDone

public void iterationDone(int iteration)

Description copied from interface: IterationListener

Event listener for each iteration

Specified by:

iterationDone in interface IterationListener