Allow user to specify training set sizes.