CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.dataSetPreProcessor(org.nd4j.linalg.dataset.api.DataSetPreProcessor dataSetPreProcessor) |
Optional DataSetPreProcessor
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.maxSentenceLength(int maxSentenceLength) |
Maximum sentence/document length.
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.minibatchSize(int minibatchSize) |
Minibatch size to use for the DataSetIterator
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.sentenceProvider(LabeledSentenceProvider labeledSentenceProvider) |
Specify how the (labelled) sentences / documents should be provided
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.sentenceProvider(LabelAwareDocumentIterator iterator,
@NonNull List<String> labels) |
Specify how the (labelled) sentences / documents should be provided
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.sentenceProvider(LabelAwareIterator iterator,
@NonNull List<String> labels) |
Specify how the (labelled) sentences / documents should be provided
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.sentenceProvider(LabelAwareSentenceIterator iterator,
@NonNull List<String> labels) |
Specify how the (labelled) sentences / documents should be provided
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.sentencesAlongHeight(boolean sentencesAlongHeight) |
If true (default): output features data with shape [minibatchSize, 1, maxSentenceLength, wordVectorSize]
If false: output features with shape [minibatchSize, 1, wordVectorSize, maxSentenceLength]
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.tokenizerFactory(TokenizerFactory tokenizerFactory) |
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.unknownWordHandling(CnnSentenceDataSetIterator.UnknownWordHandling unknownWordHandling) |
Specify how unknown words (those that don't have a word vector in the provided WordVectors instance) should be
handled.
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.useNormalizedWordVectors(boolean useNormalizedWordVectors) |
Whether normalized word vectors should be used.
|
CnnSentenceDataSetIterator.Builder |
CnnSentenceDataSetIterator.Builder.wordVectors(WordVectors wordVectors) |
Provide the WordVectors instance that should be used for training
|