# HG changeset patch
# User James Bergstra <bergstrj@iro.umontreal.ca>
# Date 1224782771 14400
# Node ID 45b3eb429c151860221c17d45bf9e226a08ac01d
# Parent  bd937e845bbb6c68e646822a67d16c03711e501c
added train_valid_test

diff -r bd937e845bbb -r 45b3eb429c15 datasets/MNIST.py
--- a/datasets/MNIST.py	Wed Oct 22 15:56:53 2008 -0400
+++ b/datasets/MNIST.py	Thu Oct 23 13:26:11 2008 -0400
@@ -29,3 +29,12 @@
     return head(n=None, path=path)
 
 
+def train_valid_test(path=None, ntrain=50000, nvalid=10000, ntest=10000):
+    all_x, all_targ = all(path=path)
+
+    train = all_x[0:ntrain], all_targ[0:ntrain]
+    valid = all_x[ntrain:ntrain+nvalid], all_targ[ntrain:ntrain+nvalid]
+    test = all_x[ntrain+nvalid:ntrain+nvalid+ntest], all_targ[ntrain+nvalid:ntrain+nvalid+ntest]
+
+    return train, valid, test
+