Added ability to evaluate a trained model on a test set