org.deeplearning4j.nn.layers.ocnn

Class OCNNParamInitializer

java.lang.Object

org.deeplearning4j.nn.params.DefaultParamInitializer

org.deeplearning4j.nn.layers.ocnn.OCNNParamInitializer

All Implemented Interfaces:

ParamInitializer

public class OCNNParamInitializer
extends DefaultParamInitializer

Param initializer for OCNNOutputLayer

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static String	K_KEY
static String	NU_KEY
static String	R_KEY
static String	V_KEY
static String	W_KEY

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

BIAS_KEY, GAIN_KEY, WEIGHT_KEY

Constructor Summary

Constructors

Constructor and Description
OCNNParamInitializer()

Method Summary

Modifier and Type	Method and Description
List<String>	biasKeys(Layer layer) Bias parameter keys given the layer configuration
protected INDArray	createWeightMatrix(NeuralNetConfiguration configuration, INDArray weightParamView, boolean initializeParameters)
Map<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 OCNNParamInitializer	getInstance()
Map<String,INDArray>	init(NeuralNetConfiguration conf, INDArray paramsView, boolean initializeParams) Initialize the parameters
boolean	isBiasParam(Layer layer, String key) Is the specified parameter a bias?
boolean	isWeightParam(Layer layer, String key) Is the specified parameter a weight?
long	numParams(Layer layer)
long	numParams(NeuralNetConfiguration conf)
List<String>	paramKeys(Layer layer) Get a list of all parameter keys given the layer configuration
List<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, hasBias, hasLayerNorm

Methods inherited from class java.lang.Object

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

Field Detail

NU_KEY

public static final String NU_KEY

See Also:

Constant Field Values

K_KEY

public static final String K_KEY

See Also:

Constant Field Values

V_KEY

public static final String V_KEY

See Also:

Constant Field Values

W_KEY

public static final String W_KEY

See Also:

Constant Field Values

R_KEY

public static final String R_KEY

See Also:

Constant Field Values

Constructor Detail

OCNNParamInitializer

public OCNNParamInitializer()

Method Detail

getInstance

public static OCNNParamInitializer getInstance()

numParams

public long numParams(NeuralNetConfiguration conf)

Specified by:

numParams in interface ParamInitializer

Overrides:

numParams in class DefaultParamInitializer

numParams

public long numParams(Layer layer)

Specified by:

numParams in interface ParamInitializer

Overrides:

numParams in class DefaultParamInitializer

paramKeys

public List<String> paramKeys(Layer layer)

Description copied from interface: ParamInitializer

Get a list of all parameter keys given the layer configuration

Specified by:

paramKeys in interface ParamInitializer

Overrides:

paramKeys in class DefaultParamInitializer

Parameters:

layer - Layer

Returns:

All parameter keys

weightKeys

public List<String> weightKeys(Layer layer)

Description copied from interface: ParamInitializer

Weight parameter keys given the layer configuration

Specified by:

weightKeys in interface ParamInitializer

Overrides:

weightKeys in class DefaultParamInitializer

Parameters:

layer - Layer

Returns:

Weight parameter keys

biasKeys

public List<String> biasKeys(Layer layer)

Description copied from interface: ParamInitializer

Bias parameter keys given the layer configuration

Specified by:

biasKeys in interface ParamInitializer

Overrides:

biasKeys in class DefaultParamInitializer

Parameters:

layer - Layer

Returns:

Bias parameter keys

isWeightParam

public boolean isWeightParam(Layer layer,
                             String key)

Description copied from interface: ParamInitializer

Is the specified parameter a weight?

Specified by:

isWeightParam in interface ParamInitializer

Overrides:

isWeightParam in class DefaultParamInitializer

Parameters:

layer - Layer

key - Key to check

Returns:

True if parameter is a weight

isBiasParam

public boolean isBiasParam(Layer layer,
                           String key)

Description copied from interface: ParamInitializer

Is the specified parameter a bias?

Specified by:

isBiasParam in interface ParamInitializer

Overrides:

isBiasParam in class DefaultParamInitializer

Parameters:

layer - Layer

key - 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: 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)

getGradientsFromFlattened

public Map<String,INDArray> getGradientsFromFlattened(NeuralNetConfiguration conf,
                                                      INDArray gradientView)

Description copied from interface: ParamInitializer

Return 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:

getGradientsFromFlattened in interface ParamInitializer

Overrides:

getGradientsFromFlattened in class DefaultParamInitializer

Parameters:

conf - Configuration

gradientView - 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

createWeightMatrix

protected INDArray createWeightMatrix(NeuralNetConfiguration configuration,
                                      INDArray weightParamView,
                                      boolean initializeParameters)

Overrides:

createWeightMatrix in class DefaultParamInitializer