Changes in train_validate_test to get training, test and CV error