org.deeplearning4j.models.paragraphvectors

Class ParagraphVectors.Builder

java.lang.Object

org.deeplearning4j.models.word2vec.Word2Vec.Builder

org.deeplearning4j.models.paragraphvectors.ParagraphVectors.Builder

Enclosing class:

ParagraphVectors

public static class ParagraphVectors.Builder
extends Word2Vec.Builder

Field Summary

Fields inherited from class org.deeplearning4j.models.word2vec.Word2Vec.Builder

batchSize, docIter, index, iter, iterations, layerSize, learningRateDecayWords, lookupTable, lr, minLearningRate, minWordFrequency, negative, sampling, saveVocab, seed, stopWords, textVectorizer, tokenizerFactory, useAdaGrad, vocabCache, window, workers

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

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

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

index

public ParagraphVectors.Builder index(InvertedIndex index)

Overrides:

index in class Word2Vec.Builder

workers

public ParagraphVectors.Builder workers(int workers)

Overrides:

workers in class Word2Vec.Builder

sampling

public ParagraphVectors.Builder sampling(double sample)

Overrides:

sampling in class Word2Vec.Builder

negativeSample

public ParagraphVectors.Builder negativeSample(double negative)

Overrides:

negativeSample in class Word2Vec.Builder

minLearningRate

public ParagraphVectors.Builder minLearningRate(double minLearningRate)

Overrides:

minLearningRate in class Word2Vec.Builder

useAdaGrad

public ParagraphVectors.Builder useAdaGrad(boolean useAdaGrad)

Overrides:

useAdaGrad in class Word2Vec.Builder

vectorizer

public ParagraphVectors.Builder vectorizer(TextVectorizer textVectorizer)

Overrides:

vectorizer in class Word2Vec.Builder

learningRateDecayWords

public ParagraphVectors.Builder learningRateDecayWords(int learningRateDecayWords)

Overrides:

learningRateDecayWords in class Word2Vec.Builder

batchSize

public ParagraphVectors.Builder batchSize(int batchSize)

Overrides:

batchSize in class Word2Vec.Builder

saveVocab

public ParagraphVectors.Builder saveVocab(boolean saveVocab)

Overrides:

saveVocab in class Word2Vec.Builder

seed

public ParagraphVectors.Builder seed(long seed)

Overrides:

seed in class Word2Vec.Builder

iterations

public ParagraphVectors.Builder iterations(int iterations)

Overrides:

iterations in class Word2Vec.Builder

learningRate

public ParagraphVectors.Builder learningRate(double lr)

Overrides:

learningRate in class Word2Vec.Builder

iterate

public ParagraphVectors.Builder iterate(DocumentIterator iter)

Overrides:

iterate in class Word2Vec.Builder

vocabCache

public ParagraphVectors.Builder vocabCache(VocabCache cache)

Overrides:

vocabCache in class Word2Vec.Builder

minWordFrequency

public ParagraphVectors.Builder minWordFrequency(int minWordFrequency)

Overrides:

minWordFrequency in class Word2Vec.Builder

tokenizerFactory

public ParagraphVectors.Builder tokenizerFactory(TokenizerFactory tokenizerFactory)

Overrides:

tokenizerFactory in class Word2Vec.Builder

layerSize

public ParagraphVectors.Builder layerSize(int layerSize)

Overrides:

layerSize in class Word2Vec.Builder

stopWords

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

Overrides:

stopWords in class Word2Vec.Builder

windowSize

public ParagraphVectors.Builder windowSize(int window)

Overrides:

windowSize in class Word2Vec.Builder

iterate

public ParagraphVectors.Builder iterate(SentenceIterator iter)

Overrides:

iterate in class Word2Vec.Builder

lookupTable

public ParagraphVectors.Builder lookupTable(WeightLookupTable lookupTable)

Overrides:

lookupTable in class Word2Vec.Builder

build

public ParagraphVectors build()

Overrides:

build in class Word2Vec.Builder