public class Word2VecModel extends hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>
Modifier and Type | Class and Description |
---|---|
static class |
Word2VecModel.Word2VecModelInfo |
static class |
Word2VecModel.Word2VecOutput |
static class |
Word2VecModel.Word2VecParameters |
Constructor and Description |
---|
Word2VecModel(water.Key selfKey,
Word2VecModel.Word2VecParameters params,
Word2VecModel.Word2VecOutput output) |
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.
|
adaptTestForTrain, adaptTestForTrain, addMetrics, addWarning, checksum_impl, cleanup_adapt, defaultThreshold, getPublishedKeys, isSupervised, remove_impl, score, score, score, score0, score0, scoreImpl, testJavaScoring, toJava, toJava, toJavaCheckTooBig, toJavaInit, toJavaNCLASSES, toJavaPredictBody, toJavaPROB, toJavaSuper
delete_and_lock, delete, delete, read_lock, read_lock, unlock_all, unlock, update, write_lock
checksum, getBinarySerializer, remove, remove, remove, remove
public Word2VecModel(water.Key selfKey, Word2VecModel.Word2VecParameters params, Word2VecModel.Word2VecOutput output)
public final Word2VecModel.Word2VecModelInfo getModelInfo()
public hex.ModelMetrics.MetricBuilder makeMetricBuilder(java.lang.String[] domain)
makeMetricBuilder
in class hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>
public double[] score0(water.fvec.Chunk[] cs, int foo, double[] data, double[] preds)
score0
in class hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>
protected double[] score0(double[] data, double[] preds)
score0
in class hex.Model<Word2VecModel,Word2VecModel.Word2VecParameters,Word2VecModel.Word2VecOutput>
public float[] transform(java.lang.String target)
target
- - String of desired wordpublic java.util.HashMap<java.lang.String,java.lang.Float> findSynonyms(java.lang.String target, int cnt)
target
- String of desired wordcnt
- Number of synonyms to findpublic void findSynonyms(float[] tarVec, int cnt)
tarVec
- word vector of a wordcnt
- number of synonyms to findpublic float cosineSimilarity(float[] target, float[] current)
target
- - a word vectorcurrent
- - a word vectorpublic void buildModelOutput()
public void delete()
delete
in class water.Lockable<Word2VecModel>