public class ModelZooWordEmbedding
extends java.lang.Object
implements ai.djl.modality.nlp.WordEmbedding, java.lang.AutoCloseable
WordEmbedding using a ZooModel.| Constructor and Description |
|---|
ModelZooWordEmbedding(ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> model)
Constructs a
ModelZooWordEmbedding. |
| Modifier and Type | Method and Description |
|---|---|
void |
close() |
ai.djl.ndarray.NDArray |
embedWord(ai.djl.ndarray.NDArray word) |
ai.djl.ndarray.NDArray |
preprocessWordToEmbed(ai.djl.ndarray.NDManager manager,
java.lang.String word) |
java.lang.String |
unembedWord(ai.djl.ndarray.NDArray word) |
boolean |
vocabularyContains(java.lang.String word) |
public ModelZooWordEmbedding(ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> model)
ModelZooWordEmbedding.model - the model for the embedding. The model's block must consist of only an Embedding<String>.public boolean vocabularyContains(java.lang.String word)
vocabularyContains in interface ai.djl.modality.nlp.WordEmbeddingpublic ai.djl.ndarray.NDArray preprocessWordToEmbed(ai.djl.ndarray.NDManager manager,
java.lang.String word)
preprocessWordToEmbed in interface ai.djl.modality.nlp.WordEmbeddingpublic ai.djl.ndarray.NDArray embedWord(ai.djl.ndarray.NDArray word)
throws ai.djl.modality.nlp.EmbeddingException
embedWord in interface ai.djl.modality.nlp.WordEmbeddingai.djl.modality.nlp.EmbeddingExceptionpublic java.lang.String unembedWord(ai.djl.ndarray.NDArray word)
unembedWord in interface ai.djl.modality.nlp.WordEmbeddingpublic void close()
close in interface java.lang.AutoCloseable