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, 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 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, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, iupdater, l1, l1Bias, l2, l2Bias, weightInit, weightNoise

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

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

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	visibleBiasInit(double visibleBiasInit)

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

nIn, nOut, units

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

activation, activation, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, updater, updater, weightInit, weightInit, weightNoise

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

build, constrainAllParameters, constrainBias, constrainWeights, dropOut, 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

Constructor Detail

Builder

public Builder()

Method Detail

lossFunction

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

visibleBiasInit

public T visibleBiasInit(double visibleBiasInit)