hex.genmodel.algos.word2vec

Class Word2VecMojoModel

java.lang.Object

hex.genmodel.GenModel

hex.genmodel.MojoModel

hex.genmodel.algos.word2vec.Word2VecMojoModel

All Implemented Interfaces:

WordEmbeddingModel, IGenModel, java.io.Serializable, IGeneratedModel

public class Word2VecMojoModel
extends MojoModel
implements WordEmbeddingModel

See Also:

Serialized Form

Field Summary

Fields inherited from class hex.genmodel.MojoModel

_balanceClasses, _category, _defaultThreshold, _modelClassDistrib, _nclasses, _nfeatures, _priorClassDistrib, _supervised, _uuid

Fields inherited from class hex.genmodel.GenModel

_domains, _names, _offsetColumn

Method Summary

Methods

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

Methods inherited from class hex.genmodel.MojoModel

getModelCategory, getUUID, isSupervised, load, load, nclasses, nfeatures

Methods inherited from class hex.genmodel.GenModel

bitSetContains, bitSetIsInRange, calibrateClassProbabilities, correctProbabilities, createAuxKey, GBM_rescale, getColIdx, getDomainValues, getDomainValues, getDomainValues, getHeader, getModelCategories, getNames, getNumClasses, getNumCols, getNumResponseClasses, getPrediction, getPredsSize, getPredsSize, getResponseIdx, getResponseName, GLM_identityInv, GLM_inverseInv, GLM_logInv, GLM_logitInv, GLM_tweedieInv, img2pixels, isAutoEncoder, isClassifier, KMeans_closest, KMeans_distance, KMeans_distance, Kmeans_preprocessData, Kmeans_preprocessData, KMeans_simplex, log_rescale, mapEnum, score0, setInput

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

getVecSize

public int getVecSize()

Description copied from interface: WordEmbeddingModel

Dimensionality of the vector space of this Word Embedding model

Specified by:

getVecSize in interface WordEmbeddingModel

Returns:

length of word embeddings

transform0

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

Description copied from interface: WordEmbeddingModel

Transforms a given a word into a word vector

Specified by:

transform0 in interface WordEmbeddingModel

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

score0

public double[] score0(double[] row,
              double[] preds)

Description copied from class: GenModel

Subclasses implement the scoring logic. The data is pre-loaded into a re-used temp array, in the order the model expects. The predictions are loaded into the re-used temp array, which is also returned. This call exactly matches the hex.Model.score0, but uses the light-weight GenModel class.

Specified by:

score0 in class GenModel