org.deeplearning4j.nn.conf.layers

Class RBM.Builder

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<T>

org.deeplearning4j.nn.conf.layers.BaseLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.BasePretrainNetwork.Builder<RBM.Builder>

org.deeplearning4j.nn.conf.layers.RBM.Builder

Enclosing class:

RBM

public static class RBM.Builder
extends BasePretrainNetwork.Builder<RBM.Builder>

Field Summary

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

lossFunction, preTrainIterations, visibleBiasInit

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

nIn, nOut

Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder

activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, epsilon, gradientNormalization, gradientNormalizationThreshold, iupdater, l1, l1Bias, l2, l2Bias, learningRate, learningRatePolicy, learningRateSchedule, momentum, momentumAfter, rho, rmsDecay, updater, weightInit

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

dropOut, layerName

Constructor Summary

Constructors

Constructor and Description
Builder()
Builder(RBM.HiddenUnit hiddenUnit, RBM.VisibleUnit visibleUnit)

Method Summary

Modifier and Type	Method and Description
RBM	build()
RBM.Builder	hiddenUnit(RBM.HiddenUnit hiddenUnit)
RBM.Builder	k(int k)
RBM.Builder	sparsity(double sparsity)
RBM.Builder	visibleUnit(RBM.VisibleUnit visibleUnit)

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

lossFunction, preTrainIterations, visibleBiasInit

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

nIn, nOut

Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder

activation, activation, activation, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, learningRate, learningRateDecayPolicy, learningRateSchedule, momentum, momentumAfter, rho, rmsDecay, updater, updater, weightInit

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

dropOut, name

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder(RBM.HiddenUnit hiddenUnit,
               RBM.VisibleUnit visibleUnit)

Builder

public Builder()

Method Detail

build

public RBM build()

Specified by:

build in class Layer.Builder<RBM.Builder>

k

public RBM.Builder k(int k)

hiddenUnit

public RBM.Builder hiddenUnit(RBM.HiddenUnit hiddenUnit)

visibleUnit

public RBM.Builder visibleUnit(RBM.VisibleUnit visibleUnit)

sparsity

public RBM.Builder sparsity(double sparsity)