org.deeplearning4j.datasets.iterator

Class DoublesDataSetIterator

java.lang.Object

org.deeplearning4j.datasets.iterator.AbstractDataSetIterator<double[]>

org.deeplearning4j.datasets.iterator.DoublesDataSetIterator

All Implemented Interfaces:

Serializable, Iterator<DataSet>, DataSetIterator

public class DoublesDataSetIterator
extends AbstractDataSetIterator<double[]>

A simple utility iterator for creating a DataSetIterator from an Iterable<Pair<double[], double[]>>. First value in pair is the features vector, second value in pair is the labels. Supports generating 2d features/labels only

Author:

raver119@gmail.com

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DoublesDataSetIterator(Iterable<org.nd4j.linalg.primitives.Pair<double[],double[]>> iterable, int batchSize)

Method Summary

Methods inherited from class org.deeplearning4j.datasets.iterator.AbstractDataSetIterator

asyncSupported, batch, fillQueue, getLabels, getPreProcessor, hasNext, inputColumns, next, next, remove, reset, resetSupported, setPreProcessor, totalOutcomes

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.Iterator

forEachRemaining

Constructor Detail

DoublesDataSetIterator

public DoublesDataSetIterator(@NonNull
                              Iterable<org.nd4j.linalg.primitives.Pair<double[],double[]>> iterable,
                              int batchSize)

Parameters:

iterable - Iterable to source data from

batchSize - Batch size for generated DataSet objects