public class LSTMParamInitializer extends Object implements ParamInitializer
Modifier and Type | Field and Description |
---|---|
static String |
DECODER_BIAS |
static String |
DECODER_WEIGHTS |
static String |
RECURRENT_WEIGHTS |
Constructor and Description |
---|
LSTMParamInitializer() |
Modifier and Type | Method and Description |
---|---|
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
|
public static final String RECURRENT_WEIGHTS
public static final String DECODER_BIAS
public static final String DECODER_WEIGHTS
public void init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params, NeuralNetConfiguration conf)
ParamInitializer
init
in interface ParamInitializer
params
- the parameters to initializeconf
- the configurationpublic void init(Map<String,org.nd4j.linalg.api.ndarray.INDArray> params, NeuralNetConfiguration conf, org.canova.api.conf.Configuration extraConf)
ParamInitializer
init
in interface ParamInitializer
params
- the params to configureconf
- the configuration to useextraConf
- an extra configuration for extensionsCopyright © 2015. All Rights Reserved.