Assignment 3 - Building a Simple Neural Network for Classification

By Evelyn Bushell

Framework Used

I used PyTorch as my model

Summary of Model

My model uses an input and output layer as well as two hidden layers. The input layer includes 784 neurons. Each hidden layer has 128 neurons. Finally, my output layer has 10 neurons.
I used a ReLu activation function for my hidden layers and a final Softmax function for the outer layer.

Text Set Accuracy

After running for 10 epochs, my model achieved 98.17% accuracy