Yes, validation should be done in `fit` (sorry, hadn't seen this yet).