Package ai.djl.modality.nlp.embedding

Contains classes that deal with word embeddings for natural language pre-processing tasks.

Interface Summary

Interface	Description
TextEmbedding	A class to manage 1-D NDArray representations of multiple words.
WordEmbedding	A class to manage 1-D NDArray representations of words.

Class Summary

Class	Description
ModelZooTextEmbedding	A WordEmbedding using a ZooModel.
SimpleTextEmbedding	A TextEmbedding that applies a WordEmbedding to each word independently.
TrainableTextEmbedding	TrainableTextEmbedding is an implementation of TextEmbedding based on TrainableWordEmbedding block.
TrainableWordEmbedding	TrainableWordEmbedding is an implementation of WordEmbedding and Embedding based on a DefaultVocabulary.
TrainableWordEmbedding.Builder	A builder for a TrainableWordEmbedding.

Exception Summary

Exception	Description
EmbeddingException	Thrown to indicate that there was some error while loading embeddings.