org.deeplearning4j.datasets.iterator

Class FloatsDataSetIterator

java.lang.Object

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

org.deeplearning4j.datasets.iterator.FloatsDataSetIterator

All Implemented Interfaces:

Serializable, Iterator<DataSet>, DataSetIterator

public class FloatsDataSetIterator
extends AbstractDataSetIterator<float[]>

A simple utility iterator for creating a DataSetIterator from an Iterable<Pair<float[], float[]>> 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
FloatsDataSetIterator(Iterable<org.nd4j.linalg.primitives.Pair<float[],float[]>> 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

FloatsDataSetIterator

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

Parameters:

iterable - Iterable to source data from

batchSize - Batch size for generated DataSet objects