org.deeplearning4j.nn.conf.layers

Class RBM

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer

org.deeplearning4j.nn.conf.layers.FeedForwardLayer

org.deeplearning4j.nn.conf.layers.BasePretrainNetwork

org.deeplearning4j.nn.conf.layers.RBM

All Implemented Interfaces:

java.io.Serializable, java.lang.Cloneable

public class RBM
extends BasePretrainNetwork

Restricted Boltzmann Machine. Markov chain with gibbs sampling. Supports the following visible units: BINARY GAUSSIAN SOFTMAX LINEAR Supports the following hidden units: RECTIFIED BINARY GAUSSIAN SOFTMAX Based on Hinton et al.'s work Great reference: http://www.iro.umontreal.ca/~lisa/publications2/index.php/publications/show/239

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	RBM.Builder
static class	RBM.HiddenUnit
static class	RBM.VisibleUnit

Field Summary

Fields

Modifier and Type	Field and Description
protected RBM.HiddenUnit	hiddenUnit
protected int	k
protected double	sparsity
protected RBM.VisibleUnit	visibleUnit

Fields inherited from class org.deeplearning4j.nn.conf.layers.BasePretrainNetwork

lossFunction, visibleBiasInit

Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer

nIn, nOut

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer

activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, layerName, learningRate, learningRateSchedule, momentum, momentumSchedule, rho, rmsDecay, updater, weightInit

Method Summary

Modifier and Type	Method and Description
ParamInitializer	initializer()
Layer	instantiate(NeuralNetConfiguration conf, java.util.Collection<IterationListener> iterationListeners, int layerIndex, org.nd4j.linalg.api.ndarray.INDArray layerParamsView, boolean initializeParams)

Methods inherited from class org.deeplearning4j.nn.conf.layers.BasePretrainNetwork

getL1ByParam, getL2ByParam, getLearningRateByParam

Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer

getOutputType, getPreProcessorForInputType, setNIn

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer

clone, getUpdaterByParam, resetLayerDefaultConfig

Methods inherited from class java.lang.Object

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

Field Detail

hiddenUnit

protected RBM.HiddenUnit hiddenUnit

visibleUnit

protected RBM.VisibleUnit visibleUnit

k

protected int k

sparsity

protected double sparsity

Method Detail

instantiate

public Layer instantiate(NeuralNetConfiguration conf,
                         java.util.Collection<IterationListener> iterationListeners,
                         int layerIndex,
                         org.nd4j.linalg.api.ndarray.INDArray layerParamsView,
                         boolean initializeParams)

Specified by:

instantiate in class Layer

initializer

public ParamInitializer initializer()

Specified by:

initializer in class Layer