twodlearn.datasets.mnist module

class twodlearn.datasets.mnist.DataSet(images, labels, fake_data=False, one_hot=False, dtype=tf.float32, reshape=True, seed=None)

Bases: tensorflow.contrib.learn.python.learn.datasets.mnist.DataSet

class twodlearn.datasets.mnist.MnistDataset(work_directory, one_hot=True, dtype=<class 'numpy.float32'>, reshape=True, valid_size=0.1)

Bases: twodlearn.datasets.base.Datasets

filename = {'test_x': 't10k-images-idx3-ubyte.gz', 'test_y': 't10k-labels-idx1-ubyte.gz', 'train_x': 'train-images-idx3-ubyte.gz', 'train_y': 'train-labels-idx1-ubyte.gz'}

src = 'http://yann.lecun.com/exdb/mnist/'

twodlearn.datasets.mnist.load(train_dir, fake_data=False, one_hot=False, reshape=True, validation_size=5000, seed=None, dataset_name='base')

twodlearn.datasets.mnist.read_datafile(data_file, one_hot=False, num_classes=10)