add optional validation prediction