org.deeplearning4j.nn.params

Class DefaultParamInitializer

java.lang.Object

org.deeplearning4j.nn.params.DefaultParamInitializer

All Implemented Interfaces:

ParamInitializer

Direct Known Subclasses:

PretrainParamInitializer, RecursiveParamInitializer

public class DefaultParamInitializer
extends Object
implements ParamInitializer

Static weight initializer with just a weight matrix and a bias

Author:

Adam Gibson

Field Summary

Fields

Modifier and Type	Field and Description
static String	BIAS_KEY
static String	WEIGHT_KEY

Constructor Summary

Constructors

Constructor and Description
DefaultParamInitializer()

Method Summary

Modifier and Type	Method and Description
protected org.nd4j.linalg.api.ndarray.INDArray	createBias(NeuralNetConfiguration conf)
protected org.nd4j.linalg.api.ndarray.INDArray	createWeightMatrix(NeuralNetConfiguration conf)
void	init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params, NeuralNetConfiguration conf) Initialize the parameters
void	init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params, NeuralNetConfiguration conf, org.canova.api.conf.Configuration extraConf) Initialization via extra parameters where necessary

Methods inherited from class java.lang.Object

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

Field Detail

WEIGHT_KEY

public static final String WEIGHT_KEY

See Also:

Constant Field Values

BIAS_KEY

public static final String BIAS_KEY

See Also:

Constant Field Values

Constructor Detail

DefaultParamInitializer

public DefaultParamInitializer()

Method Detail

init

public void init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params,
                 NeuralNetConfiguration conf)

Description copied from interface: ParamInitializer

Initialize the parameters

Specified by:

init in interface ParamInitializer

Parameters:

params - the parameters to initialize

conf - the configuration

init

public void init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params,
                 NeuralNetConfiguration conf,
                 org.canova.api.conf.Configuration extraConf)

Description copied from interface: ParamInitializer

Initialization via extra parameters where necessary

Specified by:

init in interface ParamInitializer

Parameters:

params - the params to configure

conf - the configuration to use

extraConf - an extra configuration for extensions

createBias

protected org.nd4j.linalg.api.ndarray.INDArray createBias(NeuralNetConfiguration conf)

createWeightMatrix

protected org.nd4j.linalg.api.ndarray.INDArray createWeightMatrix(NeuralNetConfiguration conf)