org.deeplearning4j.iterator.provider

Class CollectionLabeledSentenceProvider

java.lang.Object

org.deeplearning4j.iterator.provider.CollectionLabeledSentenceProvider

All Implemented Interfaces:

LabeledSentenceProvider

public class CollectionLabeledSentenceProvider
extends java.lang.Object
implements LabeledSentenceProvider

Iterate over a set of sentences/documents, where the sentences and labels are provided in lists.

Constructor Summary

Constructors

Constructor and Description
CollectionLabeledSentenceProvider(java.util.List<java.lang.String> sentences, java.util.List<java.lang.String> labelsForSentences)
CollectionLabeledSentenceProvider(java.util.List<java.lang.String> sentences, java.util.List<java.lang.String> labelsForSentences, java.util.Random rng)

Method Summary

Modifier and Type	Method and Description
java.util.List<java.lang.String>	allLabels() Return the list of labels - this also defines the class/integer label assignment order
boolean	hasNext() Are there more sentences/documents available?
Pair<java.lang.String,java.lang.String>	nextSentence()
int	numLabelClasses() Equivalent to allLabels().size()
void	reset() Reset the iterator - including shuffling the order, if necessary/appropriate
int	totalNumSentences() Return the total number of sentences, or -1 if not available

Methods inherited from class java.lang.Object

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

Constructor Detail

CollectionLabeledSentenceProvider

public CollectionLabeledSentenceProvider(@NonNull
                                         java.util.List<java.lang.String> sentences,
                                         @NonNull
                                         java.util.List<java.lang.String> labelsForSentences)

CollectionLabeledSentenceProvider

public CollectionLabeledSentenceProvider(@NonNull
                                         java.util.List<java.lang.String> sentences,
                                         @NonNull
                                         java.util.List<java.lang.String> labelsForSentences,
                                         java.util.Random rng)

Method Detail

hasNext

public boolean hasNext()

Description copied from interface: LabeledSentenceProvider

Are there more sentences/documents available?

Specified by:

hasNext in interface LabeledSentenceProvider

nextSentence

public Pair<java.lang.String,java.lang.String> nextSentence()

Specified by:

nextSentence in interface LabeledSentenceProvider

Returns:

Pair: sentence/document text and label

reset

public void reset()

Description copied from interface: LabeledSentenceProvider

Reset the iterator - including shuffling the order, if necessary/appropriate

Specified by:

reset in interface LabeledSentenceProvider

totalNumSentences

public int totalNumSentences()

Description copied from interface: LabeledSentenceProvider

Return the total number of sentences, or -1 if not available

Specified by:

totalNumSentences in interface LabeledSentenceProvider

allLabels

public java.util.List<java.lang.String> allLabels()

Description copied from interface: LabeledSentenceProvider

Return the list of labels - this also defines the class/integer label assignment order

Specified by:

allLabels in interface LabeledSentenceProvider

numLabelClasses

public int numLabelClasses()

Description copied from interface: LabeledSentenceProvider

Equivalent to allLabels().size()

Specified by:

numLabelClasses in interface LabeledSentenceProvider