Generate splits for training