public class MnistDataFetcher extends BaseDataFetcher
Modifier and Type | Field and Description |
---|---|
protected boolean |
binarize |
protected MnistManager |
man |
protected static String |
MNIST_ROOT |
static int |
NUM_EXAMPLES |
static int |
NUM_EXAMPLES_TEST |
protected int[] |
order |
protected Random |
rng |
protected boolean |
shuffle |
protected static String |
TEMP_ROOT |
protected boolean |
train |
curr, cursor, inputColumns, log, numOutcomes, totalExamples
Constructor and Description |
---|
MnistDataFetcher() |
MnistDataFetcher(boolean binarize)
Constructor telling whether to binarize the dataset or not
|
MnistDataFetcher(boolean binarize,
boolean train,
boolean shuffle,
long rngSeed) |
Modifier and Type | Method and Description |
---|---|
void |
fetch(int numExamples) |
org.nd4j.linalg.dataset.DataSet |
next() |
void |
reset() |
createInputMatrix, createOutputMatrix, createOutputVector, cursor, getLabelName, hasMore, initializeCurrFromList, inputColumns, setLabelNames, totalExamples, totalOutcomes
public static final int NUM_EXAMPLES
public static final int NUM_EXAMPLES_TEST
protected static final String TEMP_ROOT
protected static final String MNIST_ROOT
protected transient MnistManager man
protected boolean binarize
protected boolean train
protected int[] order
protected Random rng
protected boolean shuffle
public MnistDataFetcher(boolean binarize) throws IOException
binarize
- whether to binarize the dataset or notIOException
public MnistDataFetcher(boolean binarize, boolean train, boolean shuffle, long rngSeed) throws IOException
IOException
public MnistDataFetcher() throws IOException
IOException
public void fetch(int numExamples)
public void reset()
reset
in interface DataSetFetcher
reset
in class BaseDataFetcher
public org.nd4j.linalg.dataset.DataSet next()
next
in interface DataSetFetcher
next
in class BaseDataFetcher
Copyright © 2016. All Rights Reserved.