 This should be changed to calculate loss and accuracy for the test set.
