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

Author:

Adam Gibson

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(File dir)
LabelAwareFileSentenceIterator(SentencePreProcessor preProcessor, File file) Takes a single file or directory

Method Summary

Modifier and Type	Method and Description
String	currentLabel() Returns the current label for nextSentence()
List<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,
                                      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(File dir)

Method Detail

currentLabel

public 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 List<String> currentLabels()

Description copied from interface: LabelAwareSentenceIterator

For multi label problems

Specified by:

currentLabels in interface LabelAwareSentenceIterator

Returns: