Package | Description |
---|---|
org.deeplearning4j.arbiter.layers | |
org.deeplearning4j.nn.conf.layers.variational | |
org.deeplearning4j.nn.layers.variational |
Modifier and Type | Method and Description |
---|---|
VariationalAutoencoderLayerSpace.Builder |
VariationalAutoencoderLayerSpace.Builder.reconstructionDistribution(ReconstructionDistribution distribution) |
Modifier and Type | Method and Description |
---|---|
VariationalAutoencoderLayerSpace.Builder |
VariationalAutoencoderLayerSpace.Builder.reconstructionDistribution(ParameterSpace<ReconstructionDistribution> distribution) |
Modifier and Type | Class and Description |
---|---|
class |
BernoulliReconstructionDistribution
Bernoulli reconstruction distribution for variational autoencoder.
|
class |
CompositeReconstructionDistribution
CompositeReconstructionDistribution is a reconstruction distribution built from multiple other ReconstructionDistribution
instances.
|
class |
ExponentialReconstructionDistribution
Exponential reconstruction distribution.
|
class |
GaussianReconstructionDistribution
Gaussian reconstruction distribution for variational autoencoder.
|
class |
LossFunctionWrapper
LossFunctionWrapper allows training of a VAE model with a standard (possibly deterministic) neural network loss function
for the reconstruction, instead of using a
ReconstructionDistribution as would normally be done with a VAE model. |
Modifier and Type | Method and Description |
---|---|
CompositeReconstructionDistribution.Builder |
CompositeReconstructionDistribution.Builder.addDistribution(int distributionSize,
ReconstructionDistribution distribution)
Add another distribution to the composite distribution.
|
VariationalAutoencoder.Builder |
VariationalAutoencoder.Builder.reconstructionDistribution(ReconstructionDistribution distribution)
The reconstruction distribution for the data given the hidden state - i.e., P(data|Z).
|
Constructor and Description |
---|
CompositeReconstructionDistribution(int[] distributionSizes,
ReconstructionDistribution[] reconstructionDistributions,
int totalSize) |
Modifier and Type | Field and Description |
---|---|
protected ReconstructionDistribution |
VariationalAutoencoder.reconstructionDistribution |
Copyright © 2020. All rights reserved.