Subset the training data