If you’ve already created a Dataset instance for your data frame, you could use torch.utils.data.random_split 7 to create non-overlapping Datasets.
Alternatively, you could also split the data before creating the Datasets e.g. via sklearn.model_selection.train_test_split 2 or just by manually indexing the data frame.