public class MnistDataFetcher extends BaseDataFetcher
| Modifier and Type | Field and Description |
|---|---|
protected boolean |
binarize |
protected boolean |
fOrder |
protected MnistManager |
man |
protected static String |
MNIST_ROOT |
static int |
NUM_EXAMPLES |
static int |
NUM_EXAMPLES_TEST |
protected boolean |
oneIndexed |
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, totalOutcomespublic 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
protected boolean oneIndexed
protected boolean fOrder
public MnistDataFetcher(boolean binarize)
throws IOException
binarize - whether to binarize the dataset or notIOExceptionpublic MnistDataFetcher(boolean binarize,
boolean train,
boolean shuffle,
long rngSeed)
throws IOException
IOExceptionpublic MnistDataFetcher()
throws IOException
IOExceptionpublic void fetch(int numExamples)
public void reset()
reset in interface DataSetFetcherreset in class BaseDataFetcherpublic org.nd4j.linalg.dataset.DataSet next()
next in interface DataSetFetchernext in class BaseDataFetcherCopyright © 2017. All rights reserved.