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

_algoName, _balanceClasses, _category, _defaultThreshold, _h2oVersion, _modelAttributes, _modelClassDistrib, _modelDescriptor, _mojo_version, _nclasses, _nfeatures, _priorClassDistrib, _reproducibilityInformation, _supervised, _uuid

Fields inherited from class hex.genmodel.GenModel

_domains, _names, _offsetColumn, _responseColumn, _treatmentColumn

Method Summary

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, load, nclasses, nfeatures

Methods inherited from class hex.genmodel.GenModel

bitSetContains, bitSetIsInRange, calibrateClassProbabilities, convertDouble2Float, correctProbabilities, createAuxKey, features, GBM_rescale, getCategoricalEncoding, getColIdx, getDomainValues, getDomainValues, getDomainValues, getHeader, getModelCategories, getNames, getNumClasses, getNumCols, getNumResponseClasses, getOffsetName, getOrigDomainValues, getOrigNames, getOrigNumCols, getOrigProjectionArray, getOutputDomains, getOutputNames, getPrediction, getPredictionBinomial, getPredictionMultinomial, getPredsSize, getPredsSize, getResponseIdx, getResponseName, GLM_identityInv, GLM_inverseInv, GLM_logInv, GLM_logitInv, GLM_ologitInv, GLM_tweedieInv, img2pixels, internal_threadSafeInstance, isAutoEncoder, isClassifier, KMeans_closest, KMeans_distance, KMeans_distance, KMeans_distances, Kmeans_preprocessData, Kmeans_preprocessData, KMeans_simplex, log_rescale, mapEnum, nCatFeatures, requiresOffset, score0, setCats, setCats, setInput, 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