public static class Word2Vec.Builder extends Object
Modifier and Type | Field and Description |
---|---|
protected int |
batchSize |
protected DocumentIterator |
docIter |
protected InvertedIndex |
index |
protected SentenceIterator |
iter |
protected int |
iterations |
protected int |
layerSize |
protected int |
learningRateDecayWords |
protected WeightLookupTable |
lookupTable |
protected double |
lr |
protected double |
minLearningRate |
protected int |
minWordFrequency |
protected double |
negative |
protected double |
sampling |
protected boolean |
saveVocab |
protected long |
seed |
protected List<String> |
stopWords |
protected TextVectorizer |
textVectorizer |
protected TokenizerFactory |
tokenizerFactory |
protected boolean |
useAdaGrad |
protected VocabCache |
vocabCache |
protected int |
window |
protected int |
workers |
Constructor and Description |
---|
Builder() |
protected int minWordFrequency
protected int layerSize
protected SentenceIterator iter
protected int window
protected TokenizerFactory tokenizerFactory
protected VocabCache vocabCache
protected DocumentIterator docIter
protected double lr
protected int iterations
protected long seed
protected boolean saveVocab
protected int batchSize
protected int learningRateDecayWords
protected boolean useAdaGrad
protected TextVectorizer textVectorizer
protected double minLearningRate
protected double negative
protected double sampling
protected int workers
protected InvertedIndex index
protected WeightLookupTable lookupTable
public Word2Vec.Builder lookupTable(WeightLookupTable lookupTable)
public Word2Vec.Builder index(InvertedIndex index)
public Word2Vec.Builder workers(int workers)
public Word2Vec.Builder sampling(double sample)
public Word2Vec.Builder negativeSample(double negative)
public Word2Vec.Builder minLearningRate(double minLearningRate)
public Word2Vec.Builder useAdaGrad(boolean useAdaGrad)
public Word2Vec.Builder vectorizer(TextVectorizer textVectorizer)
public Word2Vec.Builder learningRateDecayWords(int learningRateDecayWords)
public Word2Vec.Builder batchSize(int batchSize)
public Word2Vec.Builder saveVocab(boolean saveVocab)
public Word2Vec.Builder seed(long seed)
public Word2Vec.Builder iterations(int iterations)
public Word2Vec.Builder learningRate(double lr)
public Word2Vec.Builder iterate(DocumentIterator iter)
public Word2Vec.Builder vocabCache(VocabCache cache)
public Word2Vec.Builder minWordFrequency(int minWordFrequency)
public Word2Vec.Builder tokenizerFactory(TokenizerFactory tokenizerFactory)
public Word2Vec.Builder layerSize(int layerSize)
public Word2Vec.Builder stopWords(List<String> stopWords)
public Word2Vec.Builder windowSize(int window)
public Word2Vec.Builder iterate(SentenceIterator iter)
public Word2Vec build()
Copyright © 2015. All rights reserved.