org.deeplearning4j.text.sentenceiterator.labelaware

Class LabelAwareFileSentenceIterator

java.lang.Object

org.deeplearning4j.text.sentenceiterator.BaseSentenceIterator

org.deeplearning4j.text.sentenceiterator.FileSentenceIterator

org.deeplearning4j.text.sentenceiterator.labelaware.LabelAwareFileSentenceIterator

All Implemented Interfaces:

LabelAwareSentenceIterator, SentenceIterator

public class LabelAwareFileSentenceIterator
extends FileSentenceIterator
implements LabelAwareSentenceIterator

Label aware sentence iterator

Field Summary

Fields inherited from class org.deeplearning4j.text.sentenceiterator.FileSentenceIterator

cache, currentFile, currLineIterator, file, fileIterator

Fields inherited from class org.deeplearning4j.text.sentenceiterator.BaseSentenceIterator

preProcessor

Constructor Summary

Constructors

Constructor and Description
LabelAwareFileSentenceIterator(java.io.File dir)
LabelAwareFileSentenceIterator(SentencePreProcessor preProcessor, java.io.File file) Takes a single file or directory

Method Summary

Modifier and Type	Method and Description
java.lang.String	currentLabel() Returns the current label for nextSentence()
java.util.List<java.lang.String>	currentLabels() For multi label problems

Methods inherited from class org.deeplearning4j.text.sentenceiterator.FileSentenceIterator

hasNext, nextSentence, reset

Methods inherited from class org.deeplearning4j.text.sentenceiterator.BaseSentenceIterator

finish, getPreProcessor, setPreProcessor

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.text.sentenceiterator.SentenceIterator

finish, getPreProcessor, hasNext, nextSentence, reset, setPreProcessor

Constructor Detail

LabelAwareFileSentenceIterator

public LabelAwareFileSentenceIterator(SentencePreProcessor preProcessor,
                                      java.io.File file)

Takes a single file or directory

Parameters:

preProcessor - the sentence pre processor

file - the file or folder to iterate over

LabelAwareFileSentenceIterator

public LabelAwareFileSentenceIterator(java.io.File dir)

Method Detail

currentLabel

public java.lang.String currentLabel()

Description copied from interface: LabelAwareSentenceIterator

Returns the current label for nextSentence()

Specified by:

currentLabel in interface LabelAwareSentenceIterator

Returns:

the label for nextSentence()

currentLabels

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

Description copied from interface: LabelAwareSentenceIterator

For multi label problems

Specified by:

currentLabels in interface LabelAwareSentenceIterator

Returns: