Mercurial > pylearn

changeset 938:d15e4f803622

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
asdf
author gdesjardins
date Sat, 22 May 2010 17:48:03 -0400
parents 594f4fda4829
children a75bf0aca18f
files pylearn/datasets/miniblocks.py
diffstat 1 files changed, 2 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/pylearn/datasets/miniblocks.py	Sat May 22 17:44:26 2010 -0400
+++ b/pylearn/datasets/miniblocks.py	Sat May 22 17:48:03 2010 -0400
@@ -5,7 +5,7 @@
 
 from pylearn.datasets import Dataset
 
-def miniblocks(reweight=None):
+def miniblocks(reweight=None, use_inverse=False):
     # If 'reweight' is not None, then it is an integer N such that each
     # sample is duplicated k times, with k taken uniformly in {1, 2, ..., N}.
     # Some adjustment is made to ensure the dataset size is a multiple of its
@@ -50,6 +50,7 @@
     set = Dataset()
     set.train = Dataset.Obj(x = input, y = target)
     set.test = Dataset.Obj(x = input, y = target)
+    set.img_shape = (4,4)
 
     return set