org.deeplearning4j.nn.params

Class PretrainParamInitializer

java.lang.Object

org.deeplearning4j.nn.params.DefaultParamInitializer

org.deeplearning4j.nn.params.PretrainParamInitializer

All Implemented Interfaces:

ParamInitializer

public class PretrainParamInitializer
extends DefaultParamInitializer

Pretrain weight initializer. Has the visible bias as well as hidden and weight matrix.

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static String	VISIBLE_BIAS_KEY

Fields inherited from class org.deeplearning4j.nn.params.DefaultParamInitializer

BIAS_KEY, WEIGHT_KEY

Constructor Summary

Constructors

Constructor and Description
PretrainParamInitializer()

Method Summary

Modifier and Type	Method and Description
static PretrainParamInitializer	getInstance()
Map<String,org.nd4j.linalg.api.ndarray.INDArray>	init(NeuralNetConfiguration conf, org.nd4j.linalg.api.ndarray.INDArray paramsView, boolean initializeParams) Initialize the parameters
int	numParams(NeuralNetConfiguration conf, boolean backprop)

Methods inherited from class org.deeplearning4j.nn.params.DefaultParamInitializer

createBias, createWeightMatrix, getGradientsFromFlattened

Methods inherited from class java.lang.Object

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

Field Detail

VISIBLE_BIAS_KEY

public static final String VISIBLE_BIAS_KEY

See Also:

Constant Field Values

Constructor Detail

PretrainParamInitializer

public PretrainParamInitializer()

Method Detail

getInstance

public static PretrainParamInitializer getInstance()

numParams

public int numParams(NeuralNetConfiguration conf,
                     boolean backprop)

Specified by:

numParams in interface ParamInitializer

Overrides:

numParams in class DefaultParamInitializer

init

public Map<String,org.nd4j.linalg.api.ndarray.INDArray> init(NeuralNetConfiguration conf,
                                                             org.nd4j.linalg.api.ndarray.INDArray paramsView,
                                                             boolean initializeParams)

Description copied from interface: ParamInitializer

Initialize the parameters

Specified by:

init in interface ParamInitializer

Overrides:

init in class DefaultParamInitializer

Parameters:

conf - the configuration

paramsView - a view of the full network (backprop) parameters

initializeParams - 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)