org.deeplearning4j.text.documentiterator.interoperability

Class DocumentIteratorConverter

java.lang.Object

org.deeplearning4j.text.documentiterator.interoperability.DocumentIteratorConverter

All Implemented Interfaces:

Iterator<LabelledDocument>, LabelAwareIterator

public class DocumentIteratorConverter
extends Object
implements LabelAwareIterator

Simple class providing compatibility between DocumentIterator/LabelAwareDocumentIterator and LabelAwareIterator

Author:

[email protected]

Field Summary

Fields

Modifier and Type	Field and Description
protected DocumentIterator	backendIterator
protected LabelsSource	generator

Constructor Summary

Constructors

Constructor and Description
DocumentIteratorConverter(DocumentIterator iterator, LabelsSource generator)
DocumentIteratorConverter(LabelAwareDocumentIterator iterator)

Method Summary

Modifier and Type	Method and Description
LabelsSource	getLabelsSource()
boolean	hasNext()
boolean	hasNextDocument()
LabelledDocument	next()
LabelledDocument	nextDocument()
protected String	readStream(InputStream stream)
void	remove()
void	reset()
void	shutdown()

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.Iterator

forEachRemaining

Field Detail

backendIterator

protected DocumentIterator backendIterator

generator

protected LabelsSource generator

Constructor Detail

DocumentIteratorConverter

public DocumentIteratorConverter(@NonNull
                                 LabelAwareDocumentIterator iterator)

DocumentIteratorConverter

public DocumentIteratorConverter(@NonNull
                                 DocumentIterator iterator,
                                 @NonNull
                                 LabelsSource generator)

Method Detail

hasNextDocument

public boolean hasNextDocument()

Specified by:

hasNextDocument in interface LabelAwareIterator

nextDocument

public LabelledDocument nextDocument()

Specified by:

nextDocument in interface LabelAwareIterator

reset

public void reset()

Specified by:

reset in interface LabelAwareIterator

hasNext

public boolean hasNext()

Specified by:

hasNext in interface Iterator<LabelledDocument>

next

public LabelledDocument next()

Specified by:

next in interface Iterator<LabelledDocument>

remove

public void remove()

Specified by:

remove in interface Iterator<LabelledDocument>

getLabelsSource

public LabelsSource getLabelsSource()

Specified by:

getLabelsSource in interface LabelAwareIterator

shutdown

public void shutdown()

Specified by:

shutdown in interface LabelAwareIterator

readStream

protected String readStream(InputStream stream)
                     throws IOException

Throws:

IOException