grechasneak/FeedforwardNetwork

This neural network was implemented to classify images in the CIFAR-10 dataset into two classes. The network has one hidden layer with a ReLu activation function and a two class cross entropy cost function at the output. The model was trained using stochastic mini-batch gradient descent with momentum optimization.

PythonMIT