Mercurial > pylearn

changeset 1471:281efa9a4463

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
icml07_loaders uses get_filepath_in_roots
author James Bergstra <bergstrj@iro.umontreal.ca>
date Wed, 18 May 2011 10:51:11 -0400
parents 94268a161925
children ddda8d93c162
files pylearn/datasets/icml07.py
diffstat 1 files changed, 4 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/pylearn/datasets/icml07.py	Wed May 18 10:50:21 2011 -0400
+++ b/pylearn/datasets/icml07.py	Wed May 18 10:51:11 2011 -0400
@@ -3,6 +3,7 @@
 import os, sys
 import numpy
 
+from config import get_filepath_in_roots
 from pylearn.io.amat import AMat
 
 class DatasetLoader(object):
@@ -68,7 +69,9 @@
         assert numpy.all(labels < self.n_classes)
         return inputs, labels
 
-def icml07_loaders(new_version=True, rootdir='.'):
+def icml07_loaders(new_version=True, rootdir=None):
+    if rootdir is None:
+        rootdir = get_filepath_in_roots('icml07data_twiki')
     rval = dict(
         mnist_basic=DatasetLoader(
             http_source='http://www.iro.umontreal.ca/~lisa/icml2007data/mnist.zip',