Merge pull request #86 from macaodha/patch-1

Solves indexing error in Python 2.7
carpedm20 · Jan 18, 2017 · 15ef89f · 15ef89f
2 parents eef3ec4 + 8cd5480
commit 15ef89f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -473,7 +473,7 @@ def load_mnist(self):
     teY = np.asarray(teY)
 
     X = np.concatenate((trX, teX), axis=0)
-    y = np.concatenate((trY, teY), axis=0)
+    y = np.concatenate((trY, teY), axis=0).astype(np.int)
 
     seed = 547
     np.random.seed(seed)