org.deeplearning4j.rbm

Class RBMOptimizer

java.lang.Object

org.deeplearning4j.optimize.NeuralNetworkOptimizer

org.deeplearning4j.rbm.RBMOptimizer

All Implemented Interfaces:

cc.mallet.optimize.Optimizable, cc.mallet.optimize.Optimizable.ByGradientValue, Serializable, NeuralNetEpochListener

public class RBMOptimizer
extends NeuralNetworkOptimizer

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from interface cc.mallet.optimize.Optimizable

cc.mallet.optimize.Optimizable.ByBatchGradient, cc.mallet.optimize.Optimizable.ByCombiningBatchGradient, cc.mallet.optimize.Optimizable.ByGISUpdate, cc.mallet.optimize.Optimizable.ByGradient, cc.mallet.optimize.Optimizable.ByGradientValue, cc.mallet.optimize.Optimizable.ByHessian, cc.mallet.optimize.Optimizable.ByValue, cc.mallet.optimize.Optimizable.ByVotedPerceptron

Field Summary

Fields

Modifier and Type	Field and Description
protected int	k
protected int	numTimesIterated

Fields inherited from class org.deeplearning4j.optimize.NeuralNetworkOptimizer

errors, extraParams, log, lr, minLearningRate, network, opt, tolerance

Constructor Summary

Constructors

Constructor and Description
RBMOptimizer(BaseNeuralNetwork network, double lr, Object[] trainingParams)

Method Summary

Methods

Modifier and Type	Method and Description
void	getValueGradient(double[] buffer)

Methods inherited from class org.deeplearning4j.optimize.NeuralNetworkOptimizer

epochDone, getErrors, getNumParameters, getParameter, getParameters, getValue, setParameter, setParameters, train

Methods inherited from class java.lang.Object

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

Field Detail

k

protected int k

numTimesIterated

protected int numTimesIterated

Constructor Detail

RBMOptimizer

public RBMOptimizer(BaseNeuralNetwork network,
            double lr,
            Object[] trainingParams)

Method Detail

getValueGradient

public void getValueGradient(double[] buffer)

Specified by:

getValueGradient in interface cc.mallet.optimize.Optimizable.ByGradientValue

Specified by:

getValueGradient in class NeuralNetworkOptimizer