Implemented the neural network prediction, which uses the trained
network.