org.deeplearning4j.nn.modelimport.keras.layers.recurrent

Class KerasSimpleRnn

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.KerasLayer

org.deeplearning4j.nn.modelimport.keras.layers.recurrent.KerasSimpleRnn

public class KerasSimpleRnn
extends KerasLayer

Imports a Keras SimpleRNN layer as a DL4J SimpleRnn layer.

Author:

Max Pumperla

Nested Class Summary

Nested classes/interfaces inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

KerasLayer.DimOrder

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	returnSequences
protected boolean	unroll

Fields inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

className, conf, dimOrder, dropout, inboundLayerNames, inputShape, kerasMajorVersion, layer, layerName, vertex, weightL1Regularization, weightL2Regularization, weights

Constructor Summary

Constructors

Constructor and Description
KerasSimpleRnn(Integer kerasVersion) Pass-through constructor from KerasLayer
KerasSimpleRnn(Map<String,Object> layerConfig) Constructor from parsed Keras layer configuration dictionary.
KerasSimpleRnn(Map<String,Object> layerConfig, boolean enforceTrainingConfig) Constructor from parsed Keras layer configuration dictionary.

Method Summary

Modifier and Type	Method and Description
InputPreProcessor	getInputPreprocessor(InputType... inputType) Gets appropriate DL4J InputPreProcessor for given InputTypes.
int	getNumParams() Returns number of trainable parameters in layer.
InputType	getOutputType(InputType... inputType) Get layer output type.
Layer	getSimpleRnnLayer() Get DL4J SimpleRnn layer.
boolean	getUnroll() Get whether SimpleRnn layer should be unrolled (for truncated BPTT).
void	setWeights(Map<String,INDArray> weights) Set weights for layer.

Methods inherited from class org.deeplearning4j.nn.modelimport.keras.KerasLayer

clearCustomLayers, clearLambdaLayers, copyWeightsToLayer, getClassName, getDimOrder, getInboundLayerNames, getInputShape, getKerasMajorVersion, getLayer, getLayerName, getNInFromConfig, getVertex, getWeights, isInputPreProcessor, isLayer, isValidInboundLayer, isVertex, registerCustomLayer, registerLambdaLayer, setInboundLayerNames, usesRegularization

Methods inherited from class java.lang.Object

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

Field Detail

unroll

protected boolean unroll

returnSequences

protected boolean returnSequences

Constructor Detail

KerasSimpleRnn

public KerasSimpleRnn(Integer kerasVersion)
               throws UnsupportedKerasConfigurationException

Pass-through constructor from KerasLayer

Parameters:

kerasVersion - major keras version

Throws:

UnsupportedKerasConfigurationException - Unsupported Keras config

KerasSimpleRnn

public KerasSimpleRnn(Map<String,Object> layerConfig)
               throws InvalidKerasConfigurationException,
                      UnsupportedKerasConfigurationException

Constructor from parsed Keras layer configuration dictionary.

Parameters:

layerConfig - dictionary containing Keras layer configuration.

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config

KerasSimpleRnn

public KerasSimpleRnn(Map<String,Object> layerConfig,
                      boolean enforceTrainingConfig)
               throws InvalidKerasConfigurationException,
                      UnsupportedKerasConfigurationException

Constructor from parsed Keras layer configuration dictionary.

Parameters:

layerConfig - dictionary containing Keras layer configuration

enforceTrainingConfig - whether to enforce training-related configuration options

Throws:

InvalidKerasConfigurationException - Invalid Keras config

UnsupportedKerasConfigurationException - Unsupported Keras config

Method Detail

getSimpleRnnLayer

public Layer getSimpleRnnLayer()

Get DL4J SimpleRnn layer.

Returns:

SimpleRnn Layer

getOutputType

public InputType getOutputType(InputType... inputType)
                        throws InvalidKerasConfigurationException

Get layer output type.

Overrides:

getOutputType in class KerasLayer

Parameters:

inputType - Array of InputTypes

Returns:

output type as InputType

Throws:

InvalidKerasConfigurationException - Invalid Keras config

getNumParams

public int getNumParams()

Returns number of trainable parameters in layer.

Overrides:

getNumParams in class KerasLayer

Returns:

number of trainable parameters (12)

getInputPreprocessor

public InputPreProcessor getInputPreprocessor(InputType... inputType)
                                       throws InvalidKerasConfigurationException

Gets appropriate DL4J InputPreProcessor for given InputTypes.

Overrides:

getInputPreprocessor in class KerasLayer

Parameters:

inputType - Array of InputTypes

Returns:

DL4J InputPreProcessor

Throws:

InvalidKerasConfigurationException - Invalid Keras configuration exception

See Also:

InputPreProcessor

getUnroll

public boolean getUnroll()

Get whether SimpleRnn layer should be unrolled (for truncated BPTT).

Returns:

whether RNN should be unrolled (boolean)

setWeights

public void setWeights(Map<String,INDArray> weights)
                throws InvalidKerasConfigurationException

Set weights for layer.

Overrides:

setWeights in class KerasLayer

Parameters:

weights - Simple RNN weights

Throws:

InvalidKerasConfigurationException - Invalid Keras configuration exception