Package org.deeplearning4j.datasets.iterator

Class RandomMultiDataSetIterator

java.lang.Object

org.deeplearning4j.datasets.iterator.RandomMultiDataSetIterator

All Implemented Interfaces:

Serializable, Iterator<MultiDataSet>, MultiDataSetIterator

public class RandomMultiDataSetIterator
extends Object
implements MultiDataSetIterator

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	RandomMultiDataSetIterator.Builder
static class	RandomMultiDataSetIterator.Values

Constructor Summary

Constructors

Constructor	Description
RandomMultiDataSetIterator(int numMiniBatches, @NonNull List<org.nd4j.common.primitives.Triple<long[],Character,RandomMultiDataSetIterator.Values>> features, @NonNull List<org.nd4j.common.primitives.Triple<long[],Character,RandomMultiDataSetIterator.Values>> labels)

Method Summary

Modifier and Type	Method	Description
boolean	asyncSupported()
static INDArray	generate(long[] shape, char order, RandomMultiDataSetIterator.Values values)	Generate a random array with the specified shape and order
static INDArray	generate(long[] shape, RandomMultiDataSetIterator.Values values)	Generate a random array with the specified shape
boolean	hasNext()
MultiDataSet	next()
MultiDataSet	next(int i)
void	remove()
void	reset()
boolean	resetSupported()

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

Methods inherited from interface org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator

getPreProcessor, setPreProcessor

Constructor Detail

RandomMultiDataSetIterator

public RandomMultiDataSetIterator(int numMiniBatches,
                                  @NonNull
                                  @NonNull List<org.nd4j.common.primitives.Triple<long[],Character,RandomMultiDataSetIterator.Values>> features,
                                  @NonNull
                                  @NonNull List<org.nd4j.common.primitives.Triple<long[],Character,RandomMultiDataSetIterator.Values>> labels)

Parameters:

numMiniBatches - Number of minibatches per epoch

features - Each triple in the list specifies the shape, array order and type of values for the features arrays

labels - Each triple in the list specifies the shape, array order and type of values for the labels arrays

Method Detail

next

public MultiDataSet next(int i)

Specified by:

next in interface MultiDataSetIterator

resetSupported

public boolean resetSupported()

Specified by:

resetSupported in interface MultiDataSetIterator

asyncSupported

public boolean asyncSupported()

Specified by:

asyncSupported in interface MultiDataSetIterator

reset

public void reset()

Specified by:

reset in interface MultiDataSetIterator

hasNext

public boolean hasNext()

Specified by:

hasNext in interface Iterator<MultiDataSet>

next

public MultiDataSet next()

Specified by:

next in interface Iterator<MultiDataSet>

remove

public void remove()

Specified by:

remove in interface Iterator<MultiDataSet>

generate

public static INDArray generate(long[] shape,
                                RandomMultiDataSetIterator.Values values)

Generate a random array with the specified shape

Parameters:

shape - Shape of the array

values - Values to fill the array with

Returns:

Random array of specified shape + contents

generate

public static INDArray generate(long[] shape,
                                char order,
                                RandomMultiDataSetIterator.Values values)

Generate a random array with the specified shape and order

Parameters:

shape - Shape of the array

order - Order of array ('c' or 'f')

values - Values to fill the array with

Returns:

Random array of specified shape + contents