nlp/com.londogard.nlp.embeddings/Embeddings

Embeddings

public interface Embeddings

Functions

public Boolean contains(String word)

Check if the word has an embedding.

public Float cosineDistance(String w1, String w2)

euclideanDistance

public Float euclideanDistance(String w1, String w2)

public abstract Character getDelimiter()

public abstract Integer getDimensions()

public abstract Map<String, NDArray<Float, D1>> getEmbeddings()

public abstract Path getFilePath()

public abstract Set<String> getVocabulary()

traverseVectors

public List<NDArray<Float, D1>> traverseVectors(List<String> words)

traverseVectorsOrNull

public List<NDArray<Float, D1>> traverseVectorsOrNull(List<String> words)

public NDArray<Float, D1> vector(String word)

Fetches Embedding if it exists for word

Inheritors

LightWordEmbeddings