Use test set in model selection