Package org.deeplearning4j.nn.params
Class VariationalAutoencoderParamInitializer
- java.lang.Object
-
- org.deeplearning4j.nn.params.DefaultParamInitializer
-
- org.deeplearning4j.nn.params.VariationalAutoencoderParamInitializer
-
- All Implemented Interfaces:
ParamInitializer
public class VariationalAutoencoderParamInitializer extends DefaultParamInitializer
-
-
Field Summary
Fields Modifier and Type Field Description static StringBIAS_KEY_SUFFIXstatic StringDECODER_PREFIXstatic StringENCODER_PREFIXstatic StringPXZ_BKey for bias parameters connecting the last decoder layer and p(data|z) (according to whateverReconstructionDistributionis set for the VAE)static StringPXZ_PREFIXstatic StringPXZ_WKey for weight parameters connecting the last decoder layer and p(data|z) (according to whateverReconstructionDistributionis set for the VAE)static StringPZX_LOGSTD2_BKey for bias parameters for log(sigma^2) in p(z|data)static StringPZX_LOGSTD2_PREFIXstatic StringPZX_LOGSTD2_WKey for weight parameters connecting the last encoder layer and the log(sigma^2) values for p(z|data)static StringPZX_MEAN_BKey for bias parameters for the mean values for p(z|data)static StringPZX_MEAN_PREFIXstatic StringPZX_MEAN_WKey for weight parameters connecting the last encoder layer and the mean values for p(z|data)static StringPZX_PREFIXstatic StringWEIGHT_KEY_SUFFIX-
Fields inherited from class org.deeplearning4j.nn.params.DefaultParamInitializer
BIAS_KEY, GAIN_KEY, WEIGHT_KEY
-
-
Constructor Summary
Constructors Constructor Description VariationalAutoencoderParamInitializer()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description List<String>biasKeys(Layer layer)Bias parameter keys given the layer configurationMap<String,INDArray>getGradientsFromFlattened(NeuralNetConfiguration conf, INDArray gradientView)Return a map of gradients (in their standard non-flattened representation), taken from the flattened (row vector) gradientView array.static VariationalAutoencoderParamInitializergetInstance()Map<String,INDArray>init(NeuralNetConfiguration conf, INDArray paramsView, boolean initializeParams)Initialize the parametersbooleanisBiasParam(Layer layer, String key)Is the specified parameter a bias?booleanisWeightParam(Layer layer, String key)Is the specified parameter a weight?longnumParams(NeuralNetConfiguration conf)List<String>paramKeys(Layer l)Get a list of all parameter keys given the layer configurationList<String>weightKeys(Layer layer)Weight parameter keys given the layer configuration-
Methods inherited from class org.deeplearning4j.nn.params.DefaultParamInitializer
createBias, createBias, createGain, createGain, createWeightMatrix, createWeightMatrix, hasBias, hasLayerNorm, numParams
-
-
-
-
Field Detail
-
WEIGHT_KEY_SUFFIX
public static final String WEIGHT_KEY_SUFFIX
- See Also:
- Constant Field Values
-
BIAS_KEY_SUFFIX
public static final String BIAS_KEY_SUFFIX
- See Also:
- Constant Field Values
-
PZX_PREFIX
public static final String PZX_PREFIX
- See Also:
- Constant Field Values
-
PZX_MEAN_PREFIX
public static final String PZX_MEAN_PREFIX
- See Also:
- Constant Field Values
-
PZX_LOGSTD2_PREFIX
public static final String PZX_LOGSTD2_PREFIX
- See Also:
- Constant Field Values
-
ENCODER_PREFIX
public static final String ENCODER_PREFIX
- See Also:
- Constant Field Values
-
DECODER_PREFIX
public static final String DECODER_PREFIX
- See Also:
- Constant Field Values
-
PZX_MEAN_W
public static final String PZX_MEAN_W
Key for weight parameters connecting the last encoder layer and the mean values for p(z|data)- See Also:
- Constant Field Values
-
PZX_MEAN_B
public static final String PZX_MEAN_B
Key for bias parameters for the mean values for p(z|data)- See Also:
- Constant Field Values
-
PZX_LOGSTD2_W
public static final String PZX_LOGSTD2_W
Key for weight parameters connecting the last encoder layer and the log(sigma^2) values for p(z|data)- See Also:
- Constant Field Values
-
PZX_LOGSTD2_B
public static final String PZX_LOGSTD2_B
Key for bias parameters for log(sigma^2) in p(z|data)- See Also:
- Constant Field Values
-
PXZ_PREFIX
public static final String PXZ_PREFIX
- See Also:
- Constant Field Values
-
PXZ_W
public static final String PXZ_W
Key for weight parameters connecting the last decoder layer and p(data|z) (according to whateverReconstructionDistributionis set for the VAE)- See Also:
- Constant Field Values
-
PXZ_B
public static final String PXZ_B
Key for bias parameters connecting the last decoder layer and p(data|z) (according to whateverReconstructionDistributionis set for the VAE)- See Also:
- Constant Field Values
-
-
Method Detail
-
getInstance
public static VariationalAutoencoderParamInitializer getInstance()
-
numParams
public long numParams(NeuralNetConfiguration conf)
- Specified by:
numParamsin interfaceParamInitializer- Overrides:
numParamsin classDefaultParamInitializer
-
paramKeys
public List<String> paramKeys(Layer l)
Description copied from interface:ParamInitializerGet a list of all parameter keys given the layer configuration- Specified by:
paramKeysin interfaceParamInitializer- Overrides:
paramKeysin classDefaultParamInitializer- Parameters:
l- Layer- Returns:
- All parameter keys
-
weightKeys
public List<String> weightKeys(Layer layer)
Description copied from interface:ParamInitializerWeight parameter keys given the layer configuration- Specified by:
weightKeysin interfaceParamInitializer- Overrides:
weightKeysin classDefaultParamInitializer- Parameters:
layer- Layer- Returns:
- Weight parameter keys
-
biasKeys
public List<String> biasKeys(Layer layer)
Description copied from interface:ParamInitializerBias parameter keys given the layer configuration- Specified by:
biasKeysin interfaceParamInitializer- Overrides:
biasKeysin classDefaultParamInitializer- Parameters:
layer- Layer- Returns:
- Bias parameter keys
-
isWeightParam
public boolean isWeightParam(Layer layer, String key)
Description copied from interface:ParamInitializerIs the specified parameter a weight?- Specified by:
isWeightParamin interfaceParamInitializer- Overrides:
isWeightParamin classDefaultParamInitializer- Parameters:
layer- Layerkey- Key to check- Returns:
- True if parameter is a weight
-
isBiasParam
public boolean isBiasParam(Layer layer, String key)
Description copied from interface:ParamInitializerIs the specified parameter a bias?- Specified by:
isBiasParamin interfaceParamInitializer- Overrides:
isBiasParamin classDefaultParamInitializer- Parameters:
layer- Layerkey- Key to check- Returns:
- True if parameter is a bias
-
init
public Map<String,INDArray> init(NeuralNetConfiguration conf, INDArray paramsView, boolean initializeParams)
Description copied from interface:ParamInitializerInitialize the parameters- Specified by:
initin interfaceParamInitializer- Overrides:
initin classDefaultParamInitializer- Parameters:
conf- the configurationparamsView- a view of the full network (backprop) parametersinitializeParams- if true: initialize the parameters according to the configuration. If false: don't modify the values in the paramsView array (but do select out the appropriate subset, reshape etc as required)- Returns:
- Map of parameters keyed by type (view of the 'paramsView' array)
-
getGradientsFromFlattened
public Map<String,INDArray> getGradientsFromFlattened(NeuralNetConfiguration conf, INDArray gradientView)
Description copied from interface:ParamInitializerReturn a map of gradients (in their standard non-flattened representation), taken from the flattened (row vector) gradientView array. The idea is that operates in exactly the same way as the paramsView does in#init(Map, NeuralNetConfiguration, INDArray); thus the position in the view (and, the array orders) must match those of the parameters- Specified by:
getGradientsFromFlattenedin interfaceParamInitializer- Overrides:
getGradientsFromFlattenedin classDefaultParamInitializer- Parameters:
conf- ConfigurationgradientView- The flattened gradients array, as a view of the larger array- Returns:
- A map containing an array by parameter type, that is a view of the full network gradients array
-
-