hex.word2vec

Class Word2VecModel

java.lang.Object

water.Iced<T>

water.Keyed<T>

water.Lockable<M>

hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>

hex.word2vec.Word2VecModel

All Implemented Interfaces:

java.io.Externalizable, java.io.Serializable, java.lang.Cloneable, water.api.StreamWriter, water.Freezable<Word2VecModel>

public class Word2VecModel
extends hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Word2VecModel.AggregateMethod
static class	Word2VecModel.Vocabulary
static class	Word2VecModel.Word2VecModelInfo
static class	Word2VecModel.Word2VecOutput
static class	Word2VecModel.Word2VecParameters
static class	Word2VecModel.WordCounts

Nested classes/interfaces inherited from class hex.Model

hex.Model.AdaptFrameParameters, hex.Model.BigScore, hex.Model.BigScoreChunkPredict, hex.Model.BigScorePredict, hex.Model.Contributions, hex.Model.DeepFeatures, hex.Model.ExemplarMembers, hex.Model.FeatureFrequencies, hex.Model.GetMostImportantFeatures, hex.Model.GetNTrees, hex.Model.GLRMArchetypes, hex.Model.GridSortBy, hex.Model.H2OModelDescriptor, hex.Model.InteractionBuilder, hex.Model.InteractionPair, hex.Model.InteractionSpec, hex.Model.JavaModelStreamWriter, hex.Model.JavaScoringOptions, hex.Model.LeafNodeAssignment, hex.Model.Output, hex.Model.Parameters, hex.Model.PredictScoreResult, hex.Model.RowToTreeAssignment, hex.Model.StagedPredictions, hex.Model.UpdateAuxTreeWeights

Field Summary

Fields inherited from class hex.Model

_dist, _input_parms, _output, _parms, _toDelete, _warnings, _warningsP, EVAL_AUTO_PARAMS_ENABLED, evalAutoParamsEnabled, scoringInfo

Fields inherited from class water.Lockable

_lockers

Fields inherited from class water.Keyed

_key

Constructor Summary

Constructors

Constructor and Description
Word2VecModel(water.Key<Word2VecModel> selfKey, Word2VecModel.Word2VecParameters params, Word2VecModel.Word2VecOutput output)

Method Summary

Modifier and Type	Method and Description
java.util.Map<java.lang.String,java.lang.Float>	findSynonyms(java.lang.String target, int cnt) Find synonyms (i.e.
Word2VecMojoWriter	getMojo()
hex.ModelMetrics.MetricBuilder	makeMetricBuilder(java.lang.String[] domain)
double[]	score0(water.fvec.Chunk[] cs, int foo, double[] data, double[] preds)
protected double[]	score0(double[] data, double[] preds)
water.fvec.Frame	toFrame() Converts this word2vec model to a Frame.
float[]	transform(java.lang.String target) Takes an input string can return the word vector for that word.
water.fvec.Frame	transform(water.fvec.Vec wordVec, Word2VecModel.AggregateMethod aggregateMethod)

Methods inherited from class hex.Model

adaptFrameForScore, adaptTestForJavaScoring, adaptTestForTrain, adaptTestForTrain, adaptTestForTrain, addMetrics, addModelMetrics, addWarning, aic, auc, AUCPR, checksum_impl, classification_error, compareTo, computeDeviances, containsResponse, data, defaultThreshold, defaultThreshold, deleteCrossValidationFoldAssignment, deleteCrossValidationModels, deleteCrossValidationPreds, deviance, deviance, evaluateAutoModelParameters, exportBinaryModel, exportMojo, fetchAll, fillScoringInfo, getDefaultGridSortBy, getGenModelEncoding, getPojoInterfaces, getToEigenVec, haveMojo, havePojo, importBinaryModel, initActualParamValues, isDistributionHuber, isFeatureUsedInPredict, isFeatureUsedInPredict, isGeneric, isSupervised, last_scored, lift_top_group, likelihood, logloss, loss, mae, makeAdaptFrameParameters, makeBigScoreTask, makeInteraction, makeInteractions, makeInteractions, makePojoWriter, makeSchema, makeScoringDomains, makeScoringNames, makeScoringNames, mean_per_class_error, modelDescriptor, mse, needsPostProcess, postProcessPredictions, predictScoreImpl, r2, readAll_impl, remove_impl, resetThreshold, result, rmsle, score, score, score, score, score, score, score, score, score0, score0, score0PostProcessSupervised, scoreMetrics, scoring_history, scoringDomains, setInputParms, setupBigScorePredict, testJavaScoring, testJavaScoring, testJavaScoring, testJavaScoring, testJavaScoring, toJava, toJava, toJava, toJavaAlgo, toJavaCheckTooBig, toJavaInit, toJavaModelClassName, toJavaPredictBody, toJavaTransform, toJavaUUID, toMojo, toMojo, toString, transform, uploadBinaryModel, writeAll_impl, writeTo

Methods inherited from class water.Lockable

delete_and_lock, delete_and_lock, delete_and_lock, delete_and_lock, delete, delete, delete, delete, read_lock, read_lock, read_lock, unlock_all, unlock, unlock, unlock, unlock, update, update, update, write_lock_to_read_lock, write_lock, write_lock, write_lock

Methods inherited from class water.Keyed

checksum_impl, checksum, checksum, getKey, readAll, remove_impl, remove_self_key_impl, remove, remove, remove, remove, remove, remove, removeQuietly, writeAll

Methods inherited from class water.Iced

asBytes, clone, copyOver, frozenType, read, readExternal, readJSON, reloadFromBytes, toJsonBytes, toJsonString, write, writeExternal, writeJSON

Methods inherited from class java.lang.Object

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

Constructor Detail

Word2VecModel

public Word2VecModel(water.Key<Word2VecModel> selfKey,
                     Word2VecModel.Word2VecParameters params,
                     Word2VecModel.Word2VecOutput output)

Method Detail

makeMetricBuilder

public hex.ModelMetrics.MetricBuilder makeMetricBuilder(java.lang.String[] domain)

Specified by:

makeMetricBuilder in class hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>

score0

public double[] score0(water.fvec.Chunk[] cs,
                       int foo,
                       double[] data,
                       double[] preds)

Overrides:

score0 in class hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>

score0

protected double[] score0(double[] data,
                          double[] preds)

Specified by:

score0 in class hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>

getMojo

public Word2VecMojoWriter getMojo()

Overrides:

getMojo in class hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>

toFrame

public water.fvec.Frame toFrame()

Converts this word2vec model to a Frame.

Returns:

Frame made of columns: Word, V1, .., Vn. Word column holds the vocabulary associated with this word2vec model, and columns V1, .., Vn represent the embeddings in the n-dimensional space.

transform

public float[] transform(java.lang.String target)

Takes an input string can return the word vector for that word.

Parameters:

target - - String of desired word

Returns:

float array containing the word vector values or null if the word isn't present in the vocabulary.

transform

public water.fvec.Frame transform(water.fvec.Vec wordVec,
                                  Word2VecModel.AggregateMethod aggregateMethod)

findSynonyms

public java.util.Map<java.lang.String,java.lang.Float> findSynonyms(java.lang.String target,
                                                                    int cnt)

Find synonyms (i.e. word-vectors with the highest cosine similarity)

Parameters:

target - String of desired word

cnt - Number of synonyms to find