org.deeplearning4j.models.word2vec

Class ParagraphVectors.Builder

java.lang.Object

org.deeplearning4j.models.word2vec.Word2Vec.Builder

org.deeplearning4j.models.word2vec.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, iter, iterations, layerSize, learningRateDecayWords, 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	build()
ParagraphVectors.Builder	trainLabels(boolean trainLabels)
ParagraphVectors.Builder	trainWords(boolean trainWords)

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

batchSize, iterate, iterate, iterations, layerSize, learningRate, learningRateDecayWords, minLearningRate, minWordFrequency, negativeSample, sampling, saveVocab, seed, stopWords, tokenizerFactory, useAdaGrad, vectorizer, vocabCache, windowSize, 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

trainWords

public ParagraphVectors.Builder trainWords(boolean trainWords)

trainLabels

public ParagraphVectors.Builder trainLabels(boolean trainLabels)

build

public ParagraphVectors build()

Overrides:

build in class Word2Vec.Builder