Update LabelTextDataset to handle cross validation