from mnist import MNIST
import numpy as np

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
	"""
	mndata = MNIST('../../resources/download')

	images = [[]] # Contains vectors of 784 pixels image
	labels = [] # Contains expected response for each image

	images, labels = mndata.load_training()

	np_images = np.array(images, dtype=np.float64)

	# Normalize data between 0.0 and 1.0
	np_images /= 255

	# Contstruct expected outputs
	np_expected = np.zeros((len(labels), 10))
	for k, label in enumerate(labels):
		np_expected[k][label] = 1.0

	return np_images, np_expected