org.deeplearning4j.streaming.conversion.dataset

Class CSVRecordToDataSet

java.lang.Object

org.deeplearning4j.streaming.conversion.dataset.CSVRecordToDataSet

All Implemented Interfaces:

java.io.Serializable, RecordToDataSet

public class CSVRecordToDataSet
extends java.lang.Object
implements RecordToDataSet

Assumes csv format and converts a batch of records in to a size() x record length matrix.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
CSVRecordToDataSet()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.dataset.DataSet	convert(java.util.Collection<java.util.Collection<org.datavec.api.writable.Writable>> records, int numLabels) Converts records in to a dataset

Methods inherited from class java.lang.Object

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

Constructor Detail

CSVRecordToDataSet

public CSVRecordToDataSet()

Method Detail

convert

public org.nd4j.linalg.dataset.DataSet convert(java.util.Collection<java.util.Collection<org.datavec.api.writable.Writable>> records,
                                               int numLabels)

Description copied from interface: RecordToDataSet

Converts records in to a dataset

Specified by:

convert in interface RecordToDataSet

Parameters:

records - the records to convert

numLabels - the number of labels for the dataset

Returns:

the converted dataset.