org.deeplearning4j.text.documentiterator

Class FileLabelAwareIterator.Builder

java.lang.Object

org.deeplearning4j.text.documentiterator.FileLabelAwareIterator.Builder

Enclosing class:

FileLabelAwareIterator

public static class FileLabelAwareIterator.Builder
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
protected List<File>	foldersToScan

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
FileLabelAwareIterator.Builder	addSourceFolder(File folder) Root folder for labels -> documents.
FileLabelAwareIterator	build()

Methods inherited from class java.lang.Object

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

Field Detail

foldersToScan

protected List<File> foldersToScan

Constructor Detail

Builder

public Builder()

Method Detail

addSourceFolder

public FileLabelAwareIterator.Builder addSourceFolder(@NonNull
                                                      File folder)

Root folder for labels -> documents. Each subfolder name will be presented as label, and contents of this folder will be represented as LabelledDocument, with label attached

Parameters:

folder - folder to be scanned for labels and files

Returns:

build

public FileLabelAwareIterator build()