Allow variable training and validation set sizes