Would it be possible to provide the dataset split ids you used for the paper, i.e. `train` `val` `test`?