org.deeplearning4j.nn.conf.layers

Class BasePretrainNetwork.Builder<T extends BasePretrainNetwork.Builder<T>>

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<T>

Direct Known Subclasses:

AutoEncoder.Builder, RBM.Builder, VariationalAutoencoder.Builder

Enclosing class:

BasePretrainNetwork

public abstract static class BasePretrainNetwork.Builder<T extends BasePretrainNetwork.Builder<T>>
extends FeedForwardLayer.Builder<T>

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction	lossFunction
protected int	preTrainIterations
protected double	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()

Method Summary

Modifier and Type	Method and Description
T	lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)
T	preTrainIterations(int preTrainIterations)
T	visibleBiasInit(double 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

build, dropOut, name

Methods inherited from class java.lang.Object

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

Field Detail

lossFunction

protected org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction

visibleBiasInit

protected double visibleBiasInit

preTrainIterations

protected int preTrainIterations

Constructor Detail

Builder

public Builder()

Method Detail

lossFunction

public T lossFunction(org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

visibleBiasInit

public T visibleBiasInit(double visibleBiasInit)

preTrainIterations

public T preTrainIterations(int preTrainIterations)