org.deeplearning4j.streaming.conversion.dataset

Interface RecordToDataSet

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

CSVRecordToDataSet

public interface RecordToDataSet
extends java.io.Serializable

Converts a list of records in to a dataset.

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

Method Detail

convert

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

Parameters:

records - the records to convert

numLabels - the number of labels for the dataset

Returns:

the converted dataset.