Package ai.djl.fasttext.zoo.nlp.word_embedding

Class FtWord2VecWordEmbedding

java.lang.Object
ai.djl.fasttext.zoo.nlp.word_embedding.FtWord2VecWordEmbedding
All Implemented Interfaces:
ai.djl.modality.nlp.embedding.WordEmbedding
public class FtWord2VecWordEmbedding extends Object implements ai.djl.modality.nlp.embedding.WordEmbedding
An implementation of WordEmbedding for FastText word embeddings.

  • Constructor Details

    • FtWord2VecWordEmbedding

      public FtWord2VecWordEmbedding(ai.djl.Model model, ai.djl.modality.nlp.Vocabulary vocabulary)
      Constructs a FtWord2VecWordEmbedding.
      Parameters:
      model - a loaded FastText wordEmbedding model or a ZooModel containing one
      vocabulary - the Vocabulary to get indices from

    • FtWord2VecWordEmbedding

      public FtWord2VecWordEmbedding(FtAbstractBlock embedding, ai.djl.modality.nlp.Vocabulary vocabulary)
      Constructs a FtWord2VecWordEmbedding.
      Parameters:
      embedding - the word embedding
      vocabulary - the Vocabulary to get indices from

  • Method Details

    • vocabularyContains

      public boolean vocabularyContains(String word)
      Specified by:
      vocabularyContains in interface ai.djl.modality.nlp.embedding.WordEmbedding

    • preprocessWordToEmbed

      public long preprocessWordToEmbed(String word)
      Specified by:
      preprocessWordToEmbed in interface ai.djl.modality.nlp.embedding.WordEmbedding

    • embedWord

      public ai.djl.ndarray.NDArray embedWord(ai.djl.ndarray.NDArray index)
      Specified by:
      embedWord in interface ai.djl.modality.nlp.embedding.WordEmbedding

    • embedWord

      public ai.djl.ndarray.NDArray embedWord(ai.djl.ndarray.NDManager manager, long index)
      Specified by:
      embedWord in interface ai.djl.modality.nlp.embedding.WordEmbedding

    • unembedWord

      public String unembedWord(ai.djl.ndarray.NDArray word)
      Specified by:
      unembedWord in interface ai.djl.modality.nlp.embedding.WordEmbedding