Keras layers

KerasEmbedding

[source]

Imports an Embedding layer from Keras.

KerasEmbedding
  1. public KerasEmbedding() throws UnsupportedKerasConfigurationException

Pass through constructor for unit tests

  • throws UnsupportedKerasConfigurationException Unsupported Keras config
getEmbeddingLayer
  1. 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
  1. 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
  1. public int getNumParams()

Returns number of trainable parameters in layer.

  • return number of trainable parameters (1)
setWeights
  1. public void setWeights(Map<String, INDArray> weights) throws InvalidKerasConfigurationException

Set weights for layer.

  • param weights Embedding layer weights