org.deeplearning4j.models.word2vec

Class Word2Vec.Builder

java.lang.Object

org.deeplearning4j.models.word2vec.Word2Vec.Builder

Enclosing class:

Word2Vec

public static class Word2Vec.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
Word2Vec.Builder	batchSize(int batchSize)
Word2Vec	build()
Word2Vec.Builder	iterate(DocumentIterator iter)
Word2Vec.Builder	iterate(SentenceIterator iter)
Word2Vec.Builder	iterations(int iterations)
Word2Vec.Builder	layerSize(int layerSize)
Word2Vec.Builder	learningRate(double lr)
Word2Vec.Builder	learningRateDecayWords(int learningRateDecayWords)
Word2Vec.Builder	minLearningRate(double minLearningRate)
Word2Vec.Builder	minWordFrequency(int minWordFrequency)
Word2Vec.Builder	negativeSample(double negative)
Word2Vec.Builder	sampling(double sample)
Word2Vec.Builder	saveVocab(boolean saveVocab)
Word2Vec.Builder	seed(long seed)
Word2Vec.Builder	stopWords(List<String> stopWords)
Word2Vec.Builder	tokenizerFactory(TokenizerFactory tokenizerFactory)
Word2Vec.Builder	useAdaGrad(boolean useAdaGrad)
Word2Vec.Builder	vectorizer(TextVectorizer textVectorizer)
Word2Vec.Builder	vocabCache(VocabCache cache)
Word2Vec.Builder	windowSize(int window)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

sampling

public Word2Vec.Builder sampling(double sample)

negativeSample

public Word2Vec.Builder negativeSample(double negative)

minLearningRate

public Word2Vec.Builder minLearningRate(double minLearningRate)

useAdaGrad

public Word2Vec.Builder useAdaGrad(boolean useAdaGrad)

vectorizer

public Word2Vec.Builder vectorizer(TextVectorizer textVectorizer)

learningRateDecayWords

public Word2Vec.Builder learningRateDecayWords(int learningRateDecayWords)

batchSize

public Word2Vec.Builder batchSize(int batchSize)

saveVocab

public Word2Vec.Builder saveVocab(boolean saveVocab)

seed

public Word2Vec.Builder seed(long seed)

iterations

public Word2Vec.Builder iterations(int iterations)

learningRate

public Word2Vec.Builder learningRate(double lr)

iterate

public Word2Vec.Builder iterate(DocumentIterator iter)

vocabCache

public Word2Vec.Builder vocabCache(VocabCache cache)

minWordFrequency

public Word2Vec.Builder minWordFrequency(int minWordFrequency)

tokenizerFactory

public Word2Vec.Builder tokenizerFactory(TokenizerFactory tokenizerFactory)

layerSize

public Word2Vec.Builder layerSize(int layerSize)

stopWords

public Word2Vec.Builder stopWords(List<String> stopWords)

windowSize

public Word2Vec.Builder windowSize(int window)

iterate

public Word2Vec.Builder iterate(SentenceIterator iter)

build

public Word2Vec build()