edflow.datasets.mnist module¶
Reference¶
-
class
edflow.datasets.mnist.
MNIST
(config=None)[source]¶ Bases:
edflow.data.dataset_mixin.DatasetMixin
-
NAME
= 'MNIST'¶
-
URL
= 'https://storage.googleapis.com/cvdf-datasets/mnist/'¶
-
FILES
= {'TEST_DATA': 't10k-images-idx3-ubyte.gz', 'TEST_LABELS': 't10k-labels-idx1-ubyte.gz', 'TRAIN_DATA': 'train-images-idx3-ubyte.gz', 'TRAIN_LABELS': 'train-labels-idx1-ubyte.gz'}¶
-
get_example
(i)[source]¶ Note
Please the documentation of
DatasetMixin
to not be confused.Add default behaviour for datasets defining an attribute
data
, which in turn is a dataset. This happens often when stacking several datasets on top of each other.The default behaviour now is to return
self.data.get_example(idx)
if possible, and otherwise revert to the original behaviour.
-
-
class
edflow.datasets.mnist.
MNISTTrain
(config=None)[source]¶ Bases:
edflow.datasets.mnist.MNIST
-
class
edflow.datasets.mnist.
MNISTTest
(config=None)[source]¶ Bases:
edflow.datasets.mnist.MNIST