org.deeplearning4j.arbiter.layers

Class VariationalAutoencoderLayerSpace.Builder

java.lang.Object

org.deeplearning4j.arbiter.layers.LayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder<T>

org.deeplearning4j.arbiter.layers.BasePretrainNetworkLayerSpace.Builder<VariationalAutoencoderLayerSpace.Builder>

org.deeplearning4j.arbiter.layers.VariationalAutoencoderLayerSpace.Builder

Enclosing class:

VariationalAutoencoderLayerSpace

public static class VariationalAutoencoderLayerSpace.Builder
extends BasePretrainNetworkLayerSpace.Builder<VariationalAutoencoderLayerSpace.Builder>

Field Summary

Fields inherited from class org.deeplearning4j.arbiter.layers.BasePretrainNetworkLayerSpace.Builder

lossFunction

Fields inherited from class org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder

constrainAll, constrainBias, constrainWeights, nIn, nOut

Fields inherited from class org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder

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

Fields inherited from class org.deeplearning4j.arbiter.layers.LayerSpace.Builder

dropOut

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
<E extends LayerSpace> E	build()
VariationalAutoencoderLayerSpace.Builder	decoderLayerSizes(int... decoderLayerSizes)
VariationalAutoencoderLayerSpace.Builder	decoderLayerSizes(ParameterSpace<int[]> decoderLayerSizes)
VariationalAutoencoderLayerSpace.Builder	encoderLayerSizes(int... encoderLayerSizes)
VariationalAutoencoderLayerSpace.Builder	encoderLayerSizes(ParameterSpace<int[]> encoderLayerSizes)
VariationalAutoencoderLayerSpace.Builder	lossFunction(org.nd4j.linalg.activations.Activation outputActivationFn, org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)
VariationalAutoencoderLayerSpace.Builder	lossFunction(org.nd4j.linalg.activations.IActivation outputActivationFn, org.nd4j.linalg.lossfunctions.ILossFunction lossFunction)
VariationalAutoencoderLayerSpace.Builder	lossFunction(org.nd4j.linalg.activations.IActivation outputActivationFn, org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)
VariationalAutoencoderLayerSpace.Builder	numSamples(int numSamples)
VariationalAutoencoderLayerSpace.Builder	numSamples(ParameterSpace<Integer> numSamples)
VariationalAutoencoderLayerSpace.Builder	pzxActivationFn(org.nd4j.linalg.activations.IActivation activationFunction)
VariationalAutoencoderLayerSpace.Builder	pzxActivationFn(ParameterSpace<org.nd4j.linalg.activations.IActivation> activationFunction)
VariationalAutoencoderLayerSpace.Builder	pzxActivationFunction(org.nd4j.linalg.activations.Activation activation)
VariationalAutoencoderLayerSpace.Builder	reconstructionDistribution(ParameterSpace<org.deeplearning4j.nn.conf.layers.variational.ReconstructionDistribution> distribution)
VariationalAutoencoderLayerSpace.Builder	reconstructionDistribution(org.deeplearning4j.nn.conf.layers.variational.ReconstructionDistribution distribution)

Methods inherited from class org.deeplearning4j.arbiter.layers.BasePretrainNetworkLayerSpace.Builder

lossFunction, lossFunction

Methods inherited from class org.deeplearning4j.arbiter.layers.FeedForwardLayerSpace.Builder

constrainAllParams, constrainAllParams, constrainBias, constrainBias, constrainWeights, constrainWeights, nIn, nIn, nOut, nOut

Methods inherited from class org.deeplearning4j.arbiter.layers.BaseLayerSpace.Builder

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

Methods inherited from class org.deeplearning4j.arbiter.layers.LayerSpace.Builder

dropOut, dropOut, iDropOut

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

encoderLayerSizes

public VariationalAutoencoderLayerSpace.Builder encoderLayerSizes(int... encoderLayerSizes)

encoderLayerSizes

public VariationalAutoencoderLayerSpace.Builder encoderLayerSizes(ParameterSpace<int[]> encoderLayerSizes)

decoderLayerSizes

public VariationalAutoencoderLayerSpace.Builder decoderLayerSizes(int... decoderLayerSizes)

decoderLayerSizes

public VariationalAutoencoderLayerSpace.Builder decoderLayerSizes(ParameterSpace<int[]> decoderLayerSizes)

reconstructionDistribution

public VariationalAutoencoderLayerSpace.Builder reconstructionDistribution(org.deeplearning4j.nn.conf.layers.variational.ReconstructionDistribution distribution)

reconstructionDistribution

public VariationalAutoencoderLayerSpace.Builder reconstructionDistribution(ParameterSpace<org.deeplearning4j.nn.conf.layers.variational.ReconstructionDistribution> distribution)

lossFunction

public VariationalAutoencoderLayerSpace.Builder lossFunction(org.nd4j.linalg.activations.IActivation outputActivationFn,
                                                             org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

lossFunction

public VariationalAutoencoderLayerSpace.Builder lossFunction(org.nd4j.linalg.activations.Activation outputActivationFn,
                                                             org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction)

lossFunction

public VariationalAutoencoderLayerSpace.Builder lossFunction(org.nd4j.linalg.activations.IActivation outputActivationFn,
                                                             org.nd4j.linalg.lossfunctions.ILossFunction lossFunction)

pzxActivationFn

public VariationalAutoencoderLayerSpace.Builder pzxActivationFn(org.nd4j.linalg.activations.IActivation activationFunction)

pzxActivationFn

public VariationalAutoencoderLayerSpace.Builder pzxActivationFn(ParameterSpace<org.nd4j.linalg.activations.IActivation> activationFunction)

pzxActivationFunction

public VariationalAutoencoderLayerSpace.Builder pzxActivationFunction(org.nd4j.linalg.activations.Activation activation)

numSamples

public VariationalAutoencoderLayerSpace.Builder numSamples(int numSamples)

numSamples

public VariationalAutoencoderLayerSpace.Builder numSamples(ParameterSpace<Integer> numSamples)

build

public <E extends LayerSpace> E build()

Specified by:

build in class LayerSpace.Builder<VariationalAutoencoderLayerSpace.Builder>