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-DNDArray
representations of multiple words.WordEmbedding A class to manage 1-DNDArray
representations of words. -
Class Summary Class Description ModelZooTextEmbedding AWordEmbedding
using aZooModel
.SimpleTextEmbedding ATextEmbedding
that applies aWordEmbedding
to each word independently.TrainableTextEmbedding TrainableWordEmbedding TrainableWordEmbedding
is an implementation ofWordEmbedding
andEmbedding
based on aDefaultVocabulary
.TrainableWordEmbedding.Builder A builder for aTrainableWordEmbedding
. -
Exception Summary Exception Description EmbeddingException Thrown to indicate that there was some error while loading embeddings.