org.deeplearning4j.datasets.iterator

Class IteratorMultiDataSetIterator

java.lang.Object

org.deeplearning4j.datasets.iterator.IteratorMultiDataSetIterator

All Implemented Interfaces:

Serializable, Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

public class IteratorMultiDataSetIterator
extends Object
implements org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

A DataSetIterator that works on an Iterator, combining and splitting the input DataSet objects as required to get a consistent batch size. Typically used in Spark training, but may be used elsewhere. NOTE: reset method is not supported here.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
IteratorMultiDataSetIterator(Iterator<org.nd4j.linalg.dataset.api.MultiDataSet> iterator, int batchSize)

Method Summary

Modifier and Type	Method and Description
boolean	asyncSupported()
org.nd4j.linalg.dataset.api.MultiDataSetPreProcessor	getPreProcessor()
boolean	hasNext()
org.nd4j.linalg.dataset.api.MultiDataSet	next()
org.nd4j.linalg.dataset.api.MultiDataSet	next(int num)
void	remove()
void	reset()
boolean	resetSupported()
void	setPreProcessor(org.nd4j.linalg.dataset.api.MultiDataSetPreProcessor preProcessor)

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

IteratorMultiDataSetIterator

public IteratorMultiDataSetIterator(Iterator<org.nd4j.linalg.dataset.api.MultiDataSet> iterator,
                                    int batchSize)

Method Detail

hasNext

public boolean hasNext()

Specified by:

hasNext in interface Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>

next

public org.nd4j.linalg.dataset.api.MultiDataSet next()

Specified by:

next in interface Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>

next

public org.nd4j.linalg.dataset.api.MultiDataSet next(int num)

Specified by:

next in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

resetSupported

public boolean resetSupported()

Specified by:

resetSupported in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

asyncSupported

public boolean asyncSupported()

Specified by:

asyncSupported in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

reset

public void reset()

Specified by:

reset in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

setPreProcessor

public void setPreProcessor(org.nd4j.linalg.dataset.api.MultiDataSetPreProcessor preProcessor)

Specified by:

setPreProcessor in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

getPreProcessor

public org.nd4j.linalg.dataset.api.MultiDataSetPreProcessor getPreProcessor()

Specified by:

getPreProcessor in interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

remove

public void remove()

Specified by:

remove in interface Iterator<org.nd4j.linalg.dataset.api.MultiDataSet>