Identifying the training set and the validation set