Package | Description |
---|---|
org.deeplearning4j.models.paragraphvectors | |
org.deeplearning4j.models.word2vec |
Modifier and Type | Class and Description |
---|---|
static class |
ParagraphVectors.Builder |
Modifier and Type | Method and Description |
---|---|
Word2Vec.Builder |
Word2Vec.Builder.batchSize(int batchSize) |
Word2Vec.Builder |
Word2Vec.Builder.epochs(int numEpochs) |
Word2Vec.Builder |
Word2Vec.Builder.hugeModelExpected(boolean reallyExpected)
If you're going for huge model built from scratches, you can use this option to avoid excessive memory use during vocab building.
|
Word2Vec.Builder |
Word2Vec.Builder.index(InvertedIndex index)
Deprecated.
|
Word2Vec.Builder |
Word2Vec.Builder.iterate(DocumentIterator iter) |
Word2Vec.Builder |
Word2Vec.Builder.iterate(SentenceIterator iter) |
Word2Vec.Builder |
Word2Vec.Builder.iterations(int iterations) |
Word2Vec.Builder |
Word2Vec.Builder.layerSize(int layerSize) |
Word2Vec.Builder |
Word2Vec.Builder.learningRate(double lr) |
Word2Vec.Builder |
Word2Vec.Builder.learningRateDecayWords(int learningRateDecayWords) |
Word2Vec.Builder |
Word2Vec.Builder.lookupTable(WeightLookupTable lookupTable) |
Word2Vec.Builder |
Word2Vec.Builder.minLearningRate(double minLearningRate) |
Word2Vec.Builder |
Word2Vec.Builder.minWordFrequency(int minWordFrequency) |
Word2Vec.Builder |
Word2Vec.Builder.negativeSample(double negative) |
Word2Vec.Builder |
Word2Vec.Builder.resetModel(boolean reset)
Checks, if model should be reset on first run, or not.
|
Word2Vec.Builder |
Word2Vec.Builder.sampling(double sample) |
Word2Vec.Builder |
Word2Vec.Builder.saveVocab(boolean saveVocab) |
Word2Vec.Builder |
Word2Vec.Builder.seed(long seed) |
Word2Vec.Builder |
Word2Vec.Builder.stopWords(List<String> stopWords) |
Word2Vec.Builder |
Word2Vec.Builder.tokenizerFactory(TokenizerFactory tokenizerFactory) |
Word2Vec.Builder |
Word2Vec.Builder.useAdaGrad(boolean useAdaGrad) |
Word2Vec.Builder |
Word2Vec.Builder.vectorizer(TextVectorizer textVectorizer)
Deprecated.
|
Word2Vec.Builder |
Word2Vec.Builder.vocabCache(VocabCache cache)
All words in this VocabCache will be treated as SPECIAL words, and they won't be affected by minWordFrequency argument.
|
Word2Vec.Builder |
Word2Vec.Builder.windowSize(int window) |
Word2Vec.Builder |
Word2Vec.Builder.workers(int workers) |
Copyright © 2015. All Rights Reserved.