nlp/com.londogard.nlp.embeddings.sentence/USifSentenceEmbeddings

USifSentenceEmbeddings

public final class USifSentenceEmbeddings implements SentenceEmbeddings

This implementation is based on the paper: Unsupervised Random Walk Sentence Embeddings: A Strong but Simple Baseline (https://www.aclweb.org/anthology/W18-3012/)

Constructors

public USifSentenceEmbeddings USifSentenceEmbeddings( Embeddings tokenEmbeddings, Map<String, Float> wordProb, Integer randomWalkLength, Integer numCommonDiscourseVector)

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