org.deeplearning4j.datasets.iterator.impl

Class BenchmarkMultiDataSetIterator

java.lang.Object

org.deeplearning4j.datasets.iterator.impl.BenchmarkMultiDataSetIterator

All Implemented Interfaces:

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

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

MultiDataset iterator for simulated inputs, or input derived from a MultiDataSet example. Primarily used for benchmarking.

Author:

Justin Long (crockpotveggies)

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
BenchmarkMultiDataSetIterator(int[][] featuresShape, int[] numLabels, int totalIterations)
BenchmarkMultiDataSetIterator(org.nd4j.linalg.dataset.MultiDataSet example, int totalIterations)

Method Summary

Modifier and Type	Method and Description
boolean	asyncSupported()
org.nd4j.linalg.dataset.api.MultiDataSetPreProcessor	getPreProcessor()
boolean	hasNext() Returns true if the iteration has more elements.
org.nd4j.linalg.dataset.MultiDataSet	next() Returns the next element in the iteration.
org.nd4j.linalg.dataset.MultiDataSet	next(int i)
void	remove() Removes from the underlying collection the last element returned by this iterator (optional operation).
void	reset()
boolean	resetSupported()
void	setPreProcessor(org.nd4j.linalg.dataset.api.MultiDataSetPreProcessor dataSetPreProcessor)

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

BenchmarkMultiDataSetIterator

public BenchmarkMultiDataSetIterator(int[][] featuresShape,
                                     int[] numLabels,
                                     int totalIterations)

BenchmarkMultiDataSetIterator

public BenchmarkMultiDataSetIterator(org.nd4j.linalg.dataset.MultiDataSet example,
                                     int totalIterations)

Method Detail

next

public org.nd4j.linalg.dataset.MultiDataSet next(int i)

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 dataSetPreProcessor)

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

hasNext

public boolean hasNext()

Returns true if the iteration has more elements. (In other words, returns true if next(int) would return an element rather than throwing an exception.)

Specified by:

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

Returns:

true if the iteration has more elements

next

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

Returns the next element in the iteration.

Specified by:

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

Returns:

the next element in the iteration

remove

public void remove()

Removes from the underlying collection the last element returned by this iterator (optional operation). This method can be called only once per call to next(int). The behavior of an iterator is unspecified if the underlying collection is modified while the iteration is in progress in any way other than by calling this method.

Specified by:

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

Throws:

UnsupportedOperationException - if the remove operation is not supported by this iterator

IllegalStateException - if the next method has not yet been called, or the remove method has already been called after the last call to the next method