This is the second file.
The data file is located here, and just sits in the same folder.
import NeuralNetwork as nn
import numpy as np
#Data collection
with np.load('mnist.npz') as data:
training_images = data['training_images']
print(training_images.shape)
training_labels = data['training_labels']
print(training_labels.shape)
layer_sizes = (784,5,10)
net = nn.NeuralNetwork(layer_sizes)
prediction = net.predict(training_images[:1])
#changing this value to a single integer does not give any problems, passing in multiple values does however. I’m thinking I need to iterate on the outside of the function rather than on the inside.
print('Prediction Shape: ')
print(prediction.shape)