transform

commit 90ac98b7cba5af11e57ae75f6ec5b4dbd33ab1c6
parent 29a37fe435f216b21cb684071597a32779abe78a
Author: Étienne Simon <esimon@esimon.eu>
Date:   Fri, 13 Jun 2014 14:00:24 +0200

Fix universe in Dataset

Diffstat:
M
dataset.py
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dataset.py b/dataset.py
@@ -18,7 +18,7 @@ class Dataset(object):
         self.load_file(prefix, 'train')
         self.load_file(prefix, 'valid')
         self.load_file(prefix, 'test')
-        self.universe = scipy.sparse.eye(self.number_entities), format='csr', dtype=theano.config.floatX)
+        self.universe = scipy.sparse.eye(self.number_entities, format='csr', dtype=theano.config.floatX)
 
     def load_file(self, prefix, name):
         with open('{0}/{1}'.format(prefix, name), 'r') as file: