weka.dl4j.iterators.provider

Class CollectionLabeledSentenceProvider

java.lang.Object

org.deeplearning4j.iterator.provider.CollectionLabeledSentenceProvider

weka.dl4j.iterators.provider.CollectionLabeledSentenceProvider

All Implemented Interfaces:

org.deeplearning4j.iterator.LabeledSentenceProvider

public class CollectionLabeledSentenceProvider
extends org.deeplearning4j.iterator.provider.CollectionLabeledSentenceProvider

Extend CollectionLabeledSentenceProvider from DL4J to support setting numClasses dynamically for regression tasks.

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
int	numLabelClasses()

Methods inherited from class org.deeplearning4j.iterator.provider.CollectionLabeledSentenceProvider

allLabels, hasNext, nextSentence, reset, totalNumSentences

Methods inherited from class java.lang.Object

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

Constructor Detail

CollectionLabeledSentenceProvider

public CollectionLabeledSentenceProvider(java.util.List<java.lang.String> sentences,
                                         java.util.List<java.lang.String> labelsForSentences,
                                         int numClasses)

Method Detail

numLabelClasses

public int numLabelClasses()

Specified by:

numLabelClasses in interface org.deeplearning4j.iterator.LabeledSentenceProvider

Overrides:

numLabelClasses in class org.deeplearning4j.iterator.provider.CollectionLabeledSentenceProvider