Package org.deeplearning4j.datasets.iterator

Class RandomMultiDataSetIterator.Builder

java.lang.Object

org.deeplearning4j.datasets.iterator.RandomMultiDataSetIterator.Builder

Enclosing class:

RandomMultiDataSetIterator

public static class RandomMultiDataSetIterator.Builder
extends Object

Constructor Summary

Constructors

Constructor	Description
Builder(int numMiniBatches)

Method Summary

Modifier and Type	Method	Description
RandomMultiDataSetIterator.Builder	addFeatures(long[] shape, char order, RandomMultiDataSetIterator.Values values)	Add a new features array to the iterator
RandomMultiDataSetIterator.Builder	addFeatures(long[] shape, RandomMultiDataSetIterator.Values values)	Add a new features array to the iterator
RandomMultiDataSetIterator.Builder	addLabels(long[] shape, char order, RandomMultiDataSetIterator.Values values)	Add a new labels array to the iterator
RandomMultiDataSetIterator.Builder	addLabels(long[] shape, RandomMultiDataSetIterator.Values values)	Add a new labels array to the iterator
RandomMultiDataSetIterator	build()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Builder

public Builder(int numMiniBatches)

Parameters:

numMiniBatches - Number of minibatches per epoch

Method Detail

addFeatures

public RandomMultiDataSetIterator.Builder addFeatures(long[] shape,
                                                      RandomMultiDataSetIterator.Values values)

Add a new features array to the iterator

Parameters:

shape - Shape of the features

values - Values to fill the array with

addFeatures

public RandomMultiDataSetIterator.Builder addFeatures(long[] shape,
                                                      char order,
                                                      RandomMultiDataSetIterator.Values values)

Add a new features array to the iterator

Parameters:

shape - Shape of the features

order - Order ('c' or 'f') for the array

values - Values to fill the array with

addLabels

public RandomMultiDataSetIterator.Builder addLabels(long[] shape,
                                                    RandomMultiDataSetIterator.Values values)

Add a new labels array to the iterator

Parameters:

shape - Shape of the features

values - Values to fill the array with

addLabels

public RandomMultiDataSetIterator.Builder addLabels(long[] shape,
                                                    char order,
                                                    RandomMultiDataSetIterator.Values values)

Add a new labels array to the iterator

Parameters:

shape - Shape of the features

order - Order ('c' or 'f') for the array

values - Values to fill the array with

build

public RandomMultiDataSetIterator build()