org.datavec.api.records.reader.impl

Class FileRecordReader

java.lang.Object

org.datavec.api.records.reader.BaseRecordReader

org.datavec.api.records.reader.impl.FileRecordReader

All Implemented Interfaces:

java.io.Closeable, java.io.Serializable, java.lang.AutoCloseable, Configurable, RecordReader

Direct Known Subclasses:

BaseCodecRecordReader, CSVSequenceRecordReader, MatlabRecordReader, RegexSequenceRecordReader, TfidfRecordReader

public class FileRecordReader
extends BaseRecordReader

File reader/writer

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	appendLabel
protected Configuration	conf
protected java.io.File	currentFile
protected InputSplit	inputSplit
protected java.util.Iterator<java.io.File>	iter
protected java.util.List<java.lang.String>	labels

Fields inherited from class org.datavec.api.records.reader.BaseRecordReader

listeners

Fields inherited from interface org.datavec.api.records.reader.RecordReader

APPEND_LABEL, LABELS, NAME_SPACE

Constructor Summary

Constructors

Constructor and Description
FileRecordReader()

Method Summary

Modifier and Type	Method and Description
void	close()
protected void	doInitialize(InputSplit split)
Configuration	getConf() Return the configuration used by this object.
int	getCurrentLabel() Return the current label.
java.util.List<java.lang.String>	getLabels() List of label strings
boolean	hasNext() Whether there are anymore records
void	initialize(Configuration conf, InputSplit split) Called once at initialization.
void	initialize(InputSplit split) Called once at initialization.
java.util.List<Record>	loadFromMetaData(java.util.List<RecordMetaData> recordMetaDatas) Load multiple records from the given a list of RecordMetaData instances
Record	loadFromMetaData(RecordMetaData recordMetaData) Load a single record from the given RecordMetaData instance Note: that for data that isn't splittable (i.e., text data that needs to be scanned/split), it is more efficient to load multiple records at once using RecordReader.loadFromMetaData(List)
java.util.List<Writable>	next() Get the next record
Record	nextRecord() Similar to RecordReader.next(), but returns a Record object, that may include metadata such as the source of the data
java.util.List<Writable>	record(java.net.URI uri, java.io.DataInputStream dataInputStream) Load the record from the given DataInputStream Unlike RecordReader.next() the internal state of the RecordReader is not modified Implementations of this method should not close the DataInputStream
void	reset() Reset record reader iterator
void	setConf(Configuration conf) Set the configuration to be used by this object.
void	setLabels(java.util.List<java.lang.String> labels)

Methods inherited from class org.datavec.api.records.reader.BaseRecordReader

getListeners, invokeListeners, setListeners, setListeners

Methods inherited from class java.lang.Object

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

Field Detail

iter

protected java.util.Iterator<java.io.File> iter

conf

protected Configuration conf

currentFile

protected java.io.File currentFile

labels

protected java.util.List<java.lang.String> labels

appendLabel

protected boolean appendLabel

inputSplit

protected InputSplit inputSplit

Constructor Detail

FileRecordReader

public FileRecordReader()

Method Detail

initialize

public void initialize(InputSplit split)
                throws java.io.IOException,
                       java.lang.InterruptedException

Description copied from interface: RecordReader

Called once at initialization.

Parameters:

split - the split that defines the range of records to read

Throws:

java.io.IOException

java.lang.InterruptedException

doInitialize

protected void doInitialize(InputSplit split)

initialize

public void initialize(Configuration conf,
                       InputSplit split)
                throws java.io.IOException,
                       java.lang.InterruptedException

Description copied from interface: RecordReader

Called once at initialization.

Parameters:

conf - a configuration for initialization

split - the split that defines the range of records to read

Throws:

java.io.IOException

java.lang.InterruptedException

next

public java.util.List<Writable> next()

Description copied from interface: RecordReader

Get the next record

Returns:

getCurrentLabel

public int getCurrentLabel()

Return the current label. The index of the current file's parent directory in the label list

Returns:

The index of the current file's parent directory

getLabels

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

Description copied from interface: RecordReader

List of label strings

Returns:

setLabels

public void setLabels(java.util.List<java.lang.String> labels)

hasNext

public boolean hasNext()

Description copied from interface: RecordReader

Whether there are anymore records

Returns:

close

public void close()
           throws java.io.IOException

Throws:

java.io.IOException

setConf

public void setConf(Configuration conf)

Description copied from interface: Configurable

Set the configuration to be used by this object.

getConf

public Configuration getConf()

Description copied from interface: Configurable

Return the configuration used by this object.

reset

public void reset()

Description copied from interface: RecordReader

Reset record reader iterator

record

public java.util.List<Writable> record(java.net.URI uri,
                                       java.io.DataInputStream dataInputStream)
                                throws java.io.IOException

Description copied from interface: RecordReader

Load the record from the given DataInputStream Unlike RecordReader.next() the internal state of the RecordReader is not modified Implementations of this method should not close the DataInputStream

Throws:

java.io.IOException - if error occurs during reading from the input stream

nextRecord

public Record nextRecord()

Description copied from interface: RecordReader

Similar to RecordReader.next(), but returns a Record object, that may include metadata such as the source of the data

Returns:

next record

loadFromMetaData

public Record loadFromMetaData(RecordMetaData recordMetaData)
                        throws java.io.IOException

Description copied from interface: RecordReader

Load a single record from the given RecordMetaData instance
Note: that for data that isn't splittable (i.e., text data that needs to be scanned/split), it is more efficient to load multiple records at once using RecordReader.loadFromMetaData(List)

Parameters:

recordMetaData - Metadata for the record that we want to load from

Returns:

Single record for the given RecordMetaData instance

Throws:

java.io.IOException - If I/O error occurs during loading

loadFromMetaData

public java.util.List<Record> loadFromMetaData(java.util.List<RecordMetaData> recordMetaDatas)
                                        throws java.io.IOException

Description copied from interface: RecordReader

Load multiple records from the given a list of RecordMetaData instances

Parameters:

recordMetaDatas - Metadata for the records that we want to load from

Returns:

Multiple records for the given RecordMetaData instances

Throws:

java.io.IOException - If I/O error occurs during loading