org.deeplearning4j.models.word2vec

Class Word2Vec

java.lang.Object

org.deeplearning4j.models.embeddings.wordvectors.WordVectorsImpl<T>

org.deeplearning4j.models.sequencevectors.SequenceVectors<VocabWord>

org.deeplearning4j.models.word2vec.Word2Vec

All Implemented Interfaces:

Serializable, WordVectors

Direct Known Subclasses:

ParagraphVectors

public class Word2Vec
extends SequenceVectors<VocabWord>

This is Word2Vec implementation based on SequenceVectors

Author:

[email protected]

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Word2Vec.Builder

Nested classes/interfaces inherited from class org.deeplearning4j.models.sequencevectors.SequenceVectors

SequenceVectors.AsyncSequencer

Field Summary

Fields

Modifier and Type	Field and Description
protected SentenceIterator	sentenceIter
protected TokenizerFactory	tokenizerFactory

Fields inherited from class org.deeplearning4j.models.sequencevectors.SequenceVectors

configuration, elementsLearningAlgorithm, eventListeners, existingModel, iterator, log, scoreElements, scoreSequences, sequenceLearningAlgorithm, unknownElement

Fields inherited from class org.deeplearning4j.models.embeddings.wordvectors.WordVectorsImpl

batchSize, DEFAULT_UNK, layerSize, learningRate, learningRateDecayWords, lookupTable, minLearningRate, minWordFrequency, modelUtils, negative, numEpochs, numIterations, resetModel, sampling, seed, stopWords, trainElementsVectors, trainSequenceVectors, useAdeGrad, useUnknown, variableWindows, vocab, window, workers

Constructor Summary

Constructors

Constructor and Description
Word2Vec()

Method Summary

Modifier and Type	Method and Description
void	setSentenceIter(SentenceIterator iterator) This method defines SentenceIterator instance, that will be used as training corpus source
void	setTokenizerFactory(TokenizerFactory tokenizerFactory) This method defines TokenizerFactory instance to be using during model building

Methods inherited from class org.deeplearning4j.models.sequencevectors.SequenceVectors

buildVocab, fit, getElementsScore, getSequencesScore, trainSequence

Methods inherited from class org.deeplearning4j.models.embeddings.wordvectors.WordVectorsImpl

accuracy, getWordVector, getWordVectorMatrix, getWordVectorMatrixNormalized, getWordVectors, getWordVectorsMean, hasWord, indexOf, lookupTable, setLookupTable, setModelUtils, setVocab, similarity, similarWordsInVocabTo, update, update, vocab, wordsNearest, wordsNearest, wordsNearest, wordsNearestSum, wordsNearestSum, wordsNearestSum

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.models.embeddings.wordvectors.WordVectors

accuracy, getUNK, getWordVector, getWordVectorMatrix, getWordVectorMatrixNormalized, getWordVectors, getWordVectorsMean, hasWord, indexOf, lookupTable, setModelUtils, setUNK, similarity, similarWordsInVocabTo, vocab, wordsNearest, wordsNearest, wordsNearest, wordsNearestSum, wordsNearestSum, wordsNearestSum

Field Detail

sentenceIter

protected transient SentenceIterator sentenceIter

tokenizerFactory

protected transient TokenizerFactory tokenizerFactory

Constructor Detail

Word2Vec

public Word2Vec()

Method Detail

setTokenizerFactory

public void setTokenizerFactory(@NonNull
                                TokenizerFactory tokenizerFactory)

This method defines TokenizerFactory instance to be using during model building

Parameters:

tokenizerFactory - TokenizerFactory instance

setSentenceIter

public void setSentenceIter(@NonNull
                            SentenceIterator iterator)

This method defines SentenceIterator instance, that will be used as training corpus source

Parameters:

iterator - SentenceIterator instance