org.deeplearning4j.text.documentiterator

Class FilenamesLabelAwareIterator

java.lang.Object

org.deeplearning4j.text.documentiterator.FilenamesLabelAwareIterator

All Implemented Interfaces:

LabelAwareIterator

public class FilenamesLabelAwareIterator
extends Object
implements LabelAwareIterator

This LabelAwareIterator scans folder for files, and returns them as LabelledDocuments. Each LabelledDocument will set it's Label to file name.

Author:

[email protected]

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	FilenamesLabelAwareIterator.Builder

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	absPath
protected List<File>	files
protected LabelsSource	labelsSource
protected AtomicInteger	position

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	FilenamesLabelAwareIterator()
protected	FilenamesLabelAwareIterator(List<File> files, LabelsSource source)

Method Summary

Modifier and Type	Method and Description
LabelsSource	getLabelsSource()
boolean	hasNextDocument()
LabelledDocument	nextDocument()
void	reset()

Methods inherited from class java.lang.Object

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

Field Detail

files

protected List<File> files

position

protected AtomicInteger position

labelsSource

protected LabelsSource labelsSource

absPath

protected boolean absPath

Constructor Detail

FilenamesLabelAwareIterator

protected FilenamesLabelAwareIterator()

FilenamesLabelAwareIterator

protected FilenamesLabelAwareIterator(@NonNull
                                      List<File> files,
                                      @NonNull
                                      LabelsSource source)

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

getLabelsSource

public LabelsSource getLabelsSource()

Specified by:

getLabelsSource in interface LabelAwareIterator