import NeuralNetwork as nn import FileReader as fr images = fr.GetImageData() labels = fr.GetLabels() imageHeight = 28 imageWidth = 28 epochs = 20 net = nn.NeuralNetwork([imageHeight * imageWidth, 51, 10], 0.2, [nn.RELU, nn.SIGMOID]) for i in range(20): print( i, net.sgdLong(images[:(50 * (i + 1))], labels[:(50 * (i + 1))], 10, nn.OPT_SGD)) net.save_weights() print(net.feed_forward(images[4])) print(labels[4])