org.deeplearning4j.models.word2vec

Class ParagraphVectors

java.lang.Object

org.deeplearning4j.models.word2vec.Word2Vec

org.deeplearning4j.models.word2vec.ParagraphVectors

All Implemented Interfaces:

Serializable, Persistable

public class ParagraphVectors
extends Word2Vec

Known in some circles as doc2vec. Officially called paragraph vectors. See: http://arxiv.org/pdf/1405.4053v2.pdf

Author:

Adam Gibson

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	ParagraphVectors.Builder

Field Summary

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

alpha, batchSize, cache, docIter, g, jobQueue, layerSize, learningRateDecayWords, log, minLearningRate, minWordFrequency, numIterations, rateOfChange, sample, saveVocab, seed, sentenceIter, serialVersionUID, shouldReset, stopWords, tokenizerFactory, topNSize, totalWords, UNK, useAdaGrad, vectorizer, window, workers

Constructor Summary

Constructors

Constructor and Description
ParagraphVectors()

Method Summary

Modifier and Type	Method and Description
void	dm(List<VocabWord> sentence, boolean trainLabels, boolean trainWords)
void	fit() Train the model

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

accuracy, addWords, buildBinaryTree, buildVocab, getCache, getLayerSize, getSentenceIter, getStopWords, getTokenizerFactory, getWindow, getWordVector, getWordVectorMatrix, getWordVectorMatrixNormalized, hasWord, indexOf, iterate, load, readStopWords, resetWeights, resetWeightsOnSetup, setCache, setLayerSize, setSentenceIter, setTokenizerFactory, setup, similarity, similarWordsInVocabTo, skipGram, trainSentence, wordsNearest, wordsNearest, wordsNearestSum, wordsNearestSum, write

Methods inherited from class java.lang.Object

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

Constructor Detail

ParagraphVectors

public ParagraphVectors()

Method Detail

dm

public void dm(List<VocabWord> sentence,
               boolean trainLabels,
               boolean trainWords)

fit

public void fit()
         throws IOException

Description copied from class: Word2Vec

Train the model

Overrides:

fit in class Word2Vec

Throws:

IOException