org.deeplearning4j.nn.modelimport.keras.layers.embeddings

Class Keras2DEmbedding

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.KerasLayer

org.deeplearning4j.nn.modelimport.keras.layers.embeddings.Keras2DEmbedding

public class Keras2DEmbedding
extends KerasLayer

Imports an Embedding layer from Keras.

Author:

dave@skymind.io

Nested Class Summary

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

KerasLayer.DimOrder

Field Summary

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

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

Constructor Summary

Constructors

Constructor and Description
Keras2DEmbedding() Pass through constructor for unit tests
Keras2DEmbedding(Map<String,Object> layerConfig) Constructor from parsed Keras layer configuration dictionary.
Keras2DEmbedding(Map<String,Object> layerConfig, boolean enforceTrainingConfig) Constructor from parsed Keras layer configuration dictionary.

Method Summary

Modifier and Type	Method and Description
EmbeddingLayer	getEmbeddingLayer() Get DL4J Embedding Sequence layer.
int	getNumParams() Returns number of trainable parameters in layer.
InputType	getOutputType(InputType... inputType) Get layer output type.
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, getInputPreprocessor, getInputShape, getKerasMajorVersion, getLayer, getLayerName, getNInFromConfig, getVertex, getWeights, isInputPreProcessor, isLayer, isValidInboundLayer, isVertex, registerCustomLayer, registerLambdaLayer, setInboundLayerNames, setLayer, usesRegularization

Methods inherited from class java.lang.Object

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

Constructor Detail

Keras2DEmbedding

public Keras2DEmbedding()
                 throws UnsupportedKerasConfigurationException

Pass through constructor for unit tests

Throws:

UnsupportedKerasConfigurationException - Unsupported Keras config

Keras2DEmbedding

public Keras2DEmbedding(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

Keras2DEmbedding

public Keras2DEmbedding(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

getEmbeddingLayer

public EmbeddingLayer getEmbeddingLayer()

Get DL4J Embedding Sequence layer.

Returns:

Embedding Sequence 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 (1)

setWeights

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

Set weights for layer.

Overrides:

setWeights in class KerasLayer

Parameters:

weights - Embedding layer weights

Throws:

InvalidKerasConfigurationException