Uses of Class
org.deeplearning4j.nn.conf.layers.variational.VariationalAutoencoder.Builder

Packages that use VariationalAutoencoder.Builder

Package	Description
org.deeplearning4j.nn.conf.layers.variational

Uses of VariationalAutoencoder.Builder in org.deeplearning4j.nn.conf.layers.variational

Methods in org.deeplearning4j.nn.conf.layers.variational that return VariationalAutoencoder.Builder

Modifier and Type	Method and Description
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.decoderLayerSizes(int... decoderLayerSizes) Size of the decoder layers, in units.
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.encoderLayerSizes(int... encoderLayerSizes) Size of the encoder layers, in units.
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.lossFunction(org.nd4j.linalg.activations.Activation outputActivationFn, org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction) Configure the VAE to use the specified loss function for the reconstruction, instead of a ReconstructionDistribution.
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.lossFunction(org.nd4j.linalg.activations.IActivation outputActivationFn, org.nd4j.linalg.lossfunctions.ILossFunction lossFunction) Configure the VAE to use the specified loss function for the reconstruction, instead of a ReconstructionDistribution.
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.lossFunction(org.nd4j.linalg.activations.IActivation outputActivationFn, org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction) Configure the VAE to use the specified loss function for the reconstruction, instead of a ReconstructionDistribution.
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.nOut(int nOut) Set the size of the VAE state Z.
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.numSamples(int numSamples) Set the number of samples per data point (from VAE state Z) used when doing pretraining.
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.pzxActivationFn(org.nd4j.linalg.activations.IActivation activationFunction) Activation function for the input to P(z|data). Care should be taken with this, as some activation functions (relu, etc) are not suitable due to being bounded in range [0,infinity).
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.pzxActivationFunction(org.nd4j.linalg.activations.Activation activation) Activation function for the input to P(z|data). Care should be taken with this, as some activation functions (relu, etc) are not suitable due to being bounded in range [0,infinity).
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.pzxActivationFunction(String activationFunction) Deprecated. Use pzxActivationFunction(Activation)
VariationalAutoencoder.Builder	VariationalAutoencoder.Builder.reconstructionDistribution(ReconstructionDistribution distribution) The reconstruction distribution for the data given the hidden state - i.e., P(data|Z). This should be selected carefully based on the type of data being modelled.