Use global rng to split dataset