Training and testing the neural network

Completed