org.deeplearning4j.bagofwords.vectorizer

Class Builder

java.lang.Object

org.deeplearning4j.bagofwords.vectorizer.Builder

Direct Known Subclasses:

BagOfWordsVectorizer.Builder, TfidfVectorizer.Builder

public abstract class Builder
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
protected int	batchSize
protected VocabCache	cache
protected DocumentIterator	docIter
protected InvertedIndex	index
protected List<String>	labels
protected int	layerSize
protected int	minWordFrequency
protected double	sample
protected SentenceIterator	sentenceIterator
protected List<String>	stopWords
protected TokenizerFactory	tokenizerFactory

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
Builder	batchSize(int batchSize)
abstract TextVectorizer	build()
Builder	cache(VocabCache cache)
Builder	index(InvertedIndex index)
Builder	iterate(DocumentIterator docIter)
Builder	iterate(SentenceIterator sentenceIterator)
Builder	labels(List<String> labels)
Builder	layerSize(int layerSize)
Builder	minWords(int minWordFrequency)
Builder	sample(double sample)
Builder	stopWords(List<String> stopWords)
Builder	tokenize(TokenizerFactory tokenizerFactory)

Methods inherited from class java.lang.Object

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

Field Detail

cache

protected VocabCache cache

tokenizerFactory

protected TokenizerFactory tokenizerFactory

stopWords

protected List<String> stopWords

layerSize

protected int layerSize

minWordFrequency

protected int minWordFrequency

docIter

protected DocumentIterator docIter

sentenceIterator

protected SentenceIterator sentenceIterator

labels

protected List<String> labels

index

protected InvertedIndex index

batchSize

protected int batchSize

sample

protected double sample

Constructor Detail

Builder

public Builder()

Method Detail

sample

public Builder sample(double sample)

batchSize

public Builder batchSize(int batchSize)

index

public Builder index(InvertedIndex index)

labels

public Builder labels(List<String> labels)

cache

public Builder cache(VocabCache cache)

tokenize

public Builder tokenize(TokenizerFactory tokenizerFactory)

stopWords

public Builder stopWords(List<String> stopWords)

layerSize

public Builder layerSize(int layerSize)

minWords

public Builder minWords(int minWordFrequency)

iterate

public Builder iterate(DocumentIterator docIter)

iterate

public Builder iterate(SentenceIterator sentenceIterator)

build

public abstract TextVectorizer build()