可以使用torch.utils.data.random_split()函数将torchvision.datasets.FashionMNIST数据集划分为训练集和测试集。该函数接受两个参数:要划分的数据集和比例列表,其中训练集所占比例在前,测试集所占比例在后。例如,将FashionMNIST数据集的80%作为训练集,20%作为测试集可以这样做:
from torch.utils.data import random_split
train_data, test_data = random_split(FashionMNIST, [80, 20])