Embedding Layers

KerasEmbedding

[source]

Imports an Embedding layer from Keras.

KerasEmbedding

public KerasEmbedding() throws UnsupportedKerasConfigurationException

Pass through constructor for unit tests

• throws UnsupportedKerasConfigurationException Unsupported Keras config

getEmbeddingLayer

public EmbeddingSequenceLayer getEmbeddingLayer()

Constructor from parsed Keras layer configuration dictionary.

• param layerConfig dictionary containing Keras layer configuration

• throws InvalidKerasConfigurationException Invalid Keras config

• throws UnsupportedKerasConfigurationException Unsupported Keras config

getOutputType

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

Get layer output type.

• param inputType Array of InputTypes

• return output type as InputType

• throws InvalidKerasConfigurationException Invalid Keras config

getNumParams

public int getNumParams()

Returns number of trainable parameters in layer.

• return number of trainable parameters (1)

setWeights

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

Set weights for layer.

• param weights Embedding layer weights