org.deeplearning4j.arbiter.data

Class MnistDataProvider

java.lang.Object

org.deeplearning4j.arbiter.data.MnistDataProvider

All Implemented Interfaces:

Serializable, DataProvider

public class MnistDataProvider
extends Object
implements DataProvider

MnistDataProvider - a DataProvider for the MNIST data set, with configurable number of epochs, batch size and RNG seed

Author:

Alex Black

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
MnistDataProvider(int numEpochs, int batchSize)
MnistDataProvider(int numEpochs, int batchSize, int rngSeed)

Method Summary

Modifier and Type	Method and Description
Class<?>	getDataType()
Object	testData(Map<String,Object> dataParameters)
Object	trainData(Map<String,Object> dataParameters)

Methods inherited from class java.lang.Object

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

Constructor Detail

MnistDataProvider

public MnistDataProvider(int numEpochs,
                         int batchSize)

MnistDataProvider

public MnistDataProvider(int numEpochs,
                         int batchSize,
                         int rngSeed)

Method Detail

trainData

public Object trainData(Map<String,Object> dataParameters)

Specified by:

trainData in interface DataProvider

testData

public Object testData(Map<String,Object> dataParameters)

Specified by:

testData in interface DataProvider

getDataType

public Class<?> getDataType()

Specified by:

getDataType in interface DataProvider