Forráskód Böngészése

Transpose outputs of sample loading

In order to keep dimensions as in the lab description.
DricomDragon 5 éve
szülő
commit
8064c8acc6
1 módosított fájl, 3 hozzáadás és 3 törlés
  1. 3 3
      python/lab/io_mnist.py

+ 3 - 3
python/lab/io_mnist.py

@@ -5,8 +5,8 @@ def load_training_samples():
 	"""
 	Return np_images, np_expected
 	where
-	np_impages is a np.array of 60 000 x 784
-	np_expected is a np.array of 60 000 x 10
+	np_impages is a np.array of 784 x 60 000
+	np_expected is a np.array of 10 x 60 000
 	"""
 	mndata = MNIST('../../resources/download')
 
@@ -25,4 +25,4 @@ def load_training_samples():
 	for k, label in enumerate(labels):
 		np_expected[k][label] = 1.0
 
-	return np_images, np_expected
+	return np.transpose(np_images), np.transpose(np_expected)