org.deeplearning4j.datasets.iterator

Class INDArrayDataSetIterator

java.lang.Object

org.deeplearning4j.datasets.iterator.AbstractDataSetIterator<INDArray>

org.deeplearning4j.datasets.iterator.INDArrayDataSetIterator

All Implemented Interfaces:

Serializable, Iterator<DataSet>, DataSetIterator

public class INDArrayDataSetIterator
extends AbstractDataSetIterator<INDArray>

A simple utility iterator for creating a DataSetIterator from an Iterable<Pair<INDArray,INDArray>>. First value in pair is the features vector, second value in pair is the labels.

Author:

raver119@gmail.com

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
INDArrayDataSetIterator(Iterable<org.nd4j.linalg.primitives.Pair<INDArray,INDArray>> 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

INDArrayDataSetIterator

public INDArrayDataSetIterator(@NonNull
                               Iterable<org.nd4j.linalg.primitives.Pair<INDArray,INDArray>> iterable,
                               int batchSize)

Parameters:

iterable - Iterable to source data from

batchSize - Batch size for generated DataSet objects