Add loss calculation

[adamhutchings]

We should be able to see the performance of the neural network on its training data. To that end, we need a function to show the neural network average loss over the testing data, using the mean squared error at first.