hex.genmodel.algos.word2vec

Interface WordEmbeddingModel

All Known Implementing Classes:

Word2VecMojoModel

public interface WordEmbeddingModel

Interface for models implementing Word Embeddings

Method Summary

Methods

Modifier and Type	Method and Description
int	getVecSize() Dimensionality of the vector space of this Word Embedding model
float[]	transform0(java.lang.String word, float[] output) Transforms a given a word into a word vector

Method Detail

getVecSize

int getVecSize()

Dimensionality of the vector space of this Word Embedding model

Returns:

length of word embeddings

transform0

float[] transform0(java.lang.String word,
                 float[] output)

Transforms a given a word into a word vector

Parameters:

word - input word

output - pre-allocated word vector embedding

Returns:

word vector embedding or null if the word is an out-of-dictionary word