Embedding Layers
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
Last modified 9mo ago