org.deeplearning4j.datasets.iterator

Class EarlyTerminationMultiDataSetIterator

java.lang.Object

org.deeplearning4j.datasets.iterator.EarlyTerminationMultiDataSetIterator

All Implemented Interfaces:

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

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

Builds an iterator that terminates once the number of minibatches returned with .next() is equal to a specified number Note that a call to .next(num) is counted as a call to return a minibatch regardless of the value of num This essentially restricts the data to this specified number of minibatches.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
EarlyTerminationMultiDataSetIterator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator underlyingIterator, int terminationPoint) Constructor takes the iterator to wrap and the number of minibatches after which the call to hasNext() will return false

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

EarlyTerminationMultiDataSetIterator

public EarlyTerminationMultiDataSetIterator(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator underlyingIterator,
                                            int terminationPoint)

Constructor takes the iterator to wrap and the number of minibatches after which the call to hasNext() will return false

Parameters:

underlyingIterator, - iterator to wrap

terminationPoint, - minibatches after which hasNext() will return false

Method Detail

next

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

Specified by:

next 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

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

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>

remove

public void remove()

Specified by:

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