convert pandas dataset to numpy array so that np.reshape can work on it

khalildh · Mar 10, 2021 · 3176809 · 3176809
1 parent f6fdfaa
commit 3176809
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/htm/examples/mnist.py b/py/htm/examples/mnist.py
@@ -39,7 +39,7 @@ def load_ds(name, num_test, shape=None):
     X = data['data']
     if shape is not None:
         new_shape = shape.insert(0, sz)
-        X = np.reshape(X, shape)
+        X = np.reshape(X.to_numpy(), shape)
 
     y = data['target'].astype(np.int32)
     # split to train/test data