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>

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
INDArrayDataSetIterator(@NonNull Iterable<org.nd4j.common.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
                               @NonNull Iterable<org.nd4j.common.primitives.Pair<INDArray,INDArray>> iterable,
                               int batchSize)

Parameters:

iterable - Iterable to source data from

batchSize - Batch size for generated DataSet objects