torch.utils.data.random_split(dataset, [train_size, test_size])
时间: 2023-11-11 08:03:29 浏览: 172
随机划分数据集train、test、val
这是 PyTorch 中的一个函数,用于将一个数据集按照指定的比例划分成训练集和测试集。其中,dataset 是要划分的数据集,train_size 和 test_size 分别是训练集和测试集的大小,可以用整数表示样本数量,也可以用小数表示占比。该函数会返回两个数据集,分别为划分后的训练集和测试集,可以通过 Python 的拆包操作进行获取。需要注意的是,划分是随机的,因此每次划分的结果可能不同。
阅读全文