I don't understand the logic here. Why not just adjust the train indices instead of introducing a second set of indices?