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.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.Word2VecModelInfo
static class	Word2VecModel.Word2VecOutput
static class	Word2VecModel.Word2VecParameters

Nested classes/interfaces inherited from class hex.Model

hex.Model.DeepFeatures, hex.Model.Output, hex.Model.Parameters

Nested classes/interfaces inherited from class water.Keyed

water.Keyed.BinarySerializer<X extends water.Keyed>

Field Summary

Fields inherited from class hex.Model

_output, _parms, _warnings

Fields inherited from class water.Lockable

_lockers

Fields inherited from class water.Keyed

_key, EMPTY_KEY_LIST

Constructor Summary

Constructors

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

Method Summary

Methods

Modifier and Type	Method and Description
void	buildModelOutput()
float	cosineSimilarity(float[] target, float[] current) Basic calculation of cosine similarity
void	delete()
void	findSynonyms(float[] tarVec, int cnt) Find synonyms (i.e.
java.util.HashMap<java.lang.String,java.lang.Float>	findSynonyms(java.lang.String target, int cnt) Find synonyms (i.e.
Word2VecModel.Word2VecModelInfo	getModelInfo()
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)
float[]	transform(java.lang.String target) Takes an input string can return the word vector for that word.

Methods inherited from class hex.Model

adaptTestForTrain, adaptTestForTrain, addMetrics, addWarning, checksum_impl, cleanup_adapt, defaultThreshold, deviance, getPublishedKeys, isSupervised, predictScoreImpl, remove_impl, score, score, score, score0, score0, scoreMetrics, testJavaScoring, toJava, toJava, toJavaCheckTooBig, toJavaInit, toJavaNCLASSES, toJavaPredictBody, toJavaPROB, toJavaSuper, toString

Methods inherited from class water.Lockable

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

Methods inherited from class water.Keyed

checksum, getBinarySerializer, remove, remove, remove, remove

Methods inherited from class water.Iced

clone, frozenType, read_impl, read, readExternal, readJSON_impl, readJSON, toJsonString, write_impl, write, writeExternal, writeJSON_impl, writeJSON

Methods inherited from class java.lang.Object

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

Constructor Detail

Word2VecModel

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

Method Detail

getModelInfo

public final Word2VecModel.Word2VecModelInfo getModelInfo()

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>

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.

findSynonyms

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

Find synonyms (i.e. wordvectors with the highest cosine similarity) of the supplied String and print them to stdout.

Parameters:

target - String of desired word

cnt - Number of synonyms to find

findSynonyms

public void findSynonyms(float[] tarVec,
                int cnt)

Find synonyms (i.e. wordvectors with the highest cosine similarity) of the word vector for a word.

Parameters:

tarVec - word vector of a word

cnt - number of synonyms to find

cosineSimilarity

public float cosineSimilarity(float[] target,
                     float[] current)

Basic calculation of cosine similarity

Parameters:

target - - a word vector

current - - a word vector

Returns:

cosine similarity between the two word vectors

buildModelOutput

public void buildModelOutput()

delete

public void delete()

Overrides:

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