Interface | Description |
---|---|
LabeledSentenceProvider |
LabeledSentenceProvider: a simple iterator interface over sentences/documents that have a label.
This is intended for use with CnnSentenceDataSetIterator |
Class | Description |
---|---|
CnnSentenceDataSetIterator |
A DataSetIterator that provides data for training a CNN sentence classification models (though can of course
be used for general documents, not just sentences.
|
CnnSentenceDataSetIterator.Builder |
Enum | Description |
---|---|
CnnSentenceDataSetIterator.Format |
Format of features:
CNN1D: For use with 1d convolution layers: Shape [minibatch, vectorSize, sentenceLength] CNN2D: For use with 2d convolution layers: Shape [minibatch, 1, vectorSize, sentenceLength] or [minibatch, 1, sentenceLength, vectorSize], depending on the setting for 'sentencesAlongHeight' configuration. |
CnnSentenceDataSetIterator.UnknownWordHandling |
Copyright © 2018. All rights reserved.