Fix tree training data shuffle bug