diff datasets/defs.py @ 211:476da2ba6a12

Add nist_P07 datasets to the predefs.
author Arnaud Bergeron <abergeron@gmail.com>
date Wed, 10 Mar 2010 09:56:02 -0500
parents f0f47b045cbf
children 4cfd0eb438af
line wrap: on
line diff
--- a/datasets/defs.py	Tue Mar 09 10:15:19 2010 -0500
+++ b/datasets/defs.py	Wed Mar 10 09:56:02 2010 -0500
@@ -1,4 +1,5 @@
-__all__ = ['nist_digits', 'nist_lower', 'nist_upper', 'nist_all', 'ocr']
+__all__ = ['nist_digits', 'nist_lower', 'nist_upper', 'nist_all', 'ocr', 
+           'nist_P07']
 
 from ftfile import FTDataSet
 import theano
@@ -35,4 +36,13 @@
                 test_data = [DATA_PATH+'ocr_test_data.ft'],
                 test_lbl = [DATA_PATH+'ocr_test_labels.ft'],
                 valid_data = [DATA_PATH+'ocr_valid_data.ft'],
-                valid_lbl = [DATA_PATH+'ocr_valid_labels.ft'])
+                valid_lbl = [DATA_PATH+'ocr_valid_labels.ft'],
+                indtype=theano.config.floatX, inscale=255.)
+
+nist_P07 = FTDataSet(train_data = [DATA_PATH+'data/P07_train'+str(i)+'_data.ft' for i in range(100)],
+                     train_lbl = [DATA_PATH+'data/P07_train'+str(i)+'_labels.ft' for i in range(100)],
+                     test_data = [DATA_PATH+'data/P07_test_data.ft'],
+                     test_lbl = [DATA_PATH+'data/P07_test_labels.ft'],
+                     valid_data = [DATA_PATH+'data/P07_valid_data.ft'],
+                     valid_lbl = [DATA_PATH+'data/P07_valid_labels.ft'],
+                     indtype=theano.config.floatX, inscale=255.)