Mercurial > pylearn

comparison pylearn/datasets/icml07.py @ 1476:8c10bda4bb5f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Configured default train/valid/test split for icml07.MNIST_rotated_background dataset. Defaults are the ones used by Hugo in the ICML07 paper and in all contracting auto-encoder papers.
author gdesjardins
date Fri, 20 May 2011 16:53:00 -0400
parents e7401822d596
children 1e4dc99a3b13
comparison
equal deleted inserted replaced
1475:e7401822d596 1476:8c10bda4bb5f
8 from pylearn.datasets.config import data_root # config 8 from pylearn.datasets.config import data_root # config
9 from pylearn.datasets.dataset import Dataset 9 from pylearn.datasets.dataset import Dataset
10 10
11 class MNIST_rotated_background(object): 11 class MNIST_rotated_background(object):
12 12
13 def __init__(self, n_train=62000, n_valid=0, n_test=0): 13 def __init__(self, n_train=10000, n_valid=2000, n_test=50000):
14 14
15 basedir = os.path.join(data_root(), 'icml07data', 'npy') 15 basedir = os.path.join(data_root(), 'icml07data', 'npy')
16 16
17 x_all = numpy.load(os.path.join(basedir, 'mnist_rotated_background_images_inputs.npy')) 17 x_all = numpy.load(os.path.join(basedir, 'mnist_rotated_background_images_inputs.npy'))
18 y_all = numpy.load(os.path.join(basedir, 'mnist_rotated_background_images_labels.npy')) 18 y_all = numpy.load(os.path.join(basedir, 'mnist_rotated_background_images_labels.npy'))