neural_network: Contains the network architecture. train.py: Use it to train the network. test.py: Use the file to test the trained network.
Train the network: python train.py
Test the network: python test.py idx idx: index of image in dataset to be used to test the network.