pytorch重写mydataset数据，这个过程如何划分训练接和测试集

我用pytorch重写mydataset数据，制作dataloader，但是这个数据集我想划分出训练集和测试集
应该在mydataset类里面写还是外面写？