Mercurial > pylearn

comparison pylearn/datasets/flickr.py @ 614:f6c74f34cd35

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Use usual data_root
author Pascal Lamblin <lamblinp@iro.umontreal.ca>
date Sat, 17 Jan 2009 00:34:14 -0500
parents 28f7dc848efc
children 4a7d413c3425
comparison
equal deleted inserted replaced
613:7261e7d6368d 614:f6c74f34cd35
5 5
6 import os 6 import os
7 import numpy 7 import numpy
8 8
9 from ..io import filetensor 9 from ..io import filetensor
10 if 0: 10 from .config import data_root
11 from .config import data_root
12 else:
13 def data_root():
14 return '/u/lisa/db/flickr/filetensor'
15 from .dataset import Dataset 11 from .dataset import Dataset
16 12
17 13
18 path_test_10class ='flickr_10classes_test.ft' 14 path_test_10class ='flickr_10classes_test.ft'
19 15
23 19
24 def basic_10class(folder = None): 20 def basic_10class(folder = None):
25 """Return the basic flickr image classification problem. 21 """Return the basic flickr image classification problem.
26 The images are 75x75, and there are 7500 training examples. 22 The images are 75x75, and there are 7500 training examples.
27 """ 23 """
28 root = data_root() if folder is None else folder 24 root = os.path.join(data_root(), 'flickr') if folder is None else folder
29 train = filetensor.read(open(os.path.join(root, path_train_10class))) 25 train = filetensor.read(open(os.path.join(root, path_train_10class)))
30 valid = filetensor.read(open(os.path.join(root, path_valid_10class))) 26 valid = filetensor.read(open(os.path.join(root, path_valid_10class)))
31 test = filetensor.read(open(os.path.join(root, path_test_10class))) 27 test = filetensor.read(open(os.path.join(root, path_test_10class)))
32 28
33 rval = Dataset() 29 rval = Dataset()