Mercurial > pylearn
view pylearn/datasets/caltech.py @ 1531:88f361283a19 tip
Fix url/name to pylearn2.
author | Frederic Bastien <nouiz@nouiz.org> |
---|---|
date | Mon, 09 Sep 2013 10:08:05 -0400 |
parents | f7b348e6a98e |
children |
line wrap: on
line source
""" Various routines to load/access MNIST data. """ import os import numpy from pylearn.io.pmat import PMat from pylearn.datasets.dataset import Dataset import config def caltech_silhouette(): rval = Dataset() # dataset lookup through $PYLEARN_DATA_ROOT path = config.get_filepath_in_roots('caltech_silhouettes') rval.train = Dataset.Obj(x=numpy.load(os.path.join(path,'train_data.npy')), y=numpy.load(os.path.join(path,'train_labels.npy'))) rval.valid = Dataset.Obj(x=numpy.load(os.path.join(path,'val_data.npy')), y=numpy.load(os.path.join(path,'val_labels.npy'))) rval.test = Dataset.Obj(x=numpy.load(os.path.join(path,'test_data.npy')), y=numpy.load(os.path.join(path,'test_labels.npy'))) rval.n_classes = 101 rval.img_shape = (28,28) return rval