nlp/com.londogard.nlp.embeddings.sentence/AverageSentenceEmbeddings

AverageSentenceEmbeddings

public final class AverageSentenceEmbeddings implements SentenceEmbeddings

Simply applies the average of vectors in sentence to build the sentence embedding.

Constructors

AverageSentenceEmbeddings

public AverageSentenceEmbeddings AverageSentenceEmbeddings(Embeddings tokenEmbeddings)

Functions

getSentenceEmbedding

public NDArray<Float, D1> getSentenceEmbedding(List<String> sentence)

getSentenceEmbeddings

public List<NDArray<Float, D1>> getSentenceEmbeddings(List<List<String>> listOfSentences)

getTokenEmbeddings

public Embeddings getTokenEmbeddings()

Properties

tokenEmbeddings

private final Embeddings tokenEmbeddings