Links

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