train, test= torch.utils.data.random_split(dataset=all_dataset, lengths=[1800,200])用这种方法对数据进行划分划分后的样本数量分布是如何的
时间: 2024-03-29 19:39:18 浏览: 106
随机划分数据集train、test、val
使用 `torch.utils.data.random_split` 划分数据集时,参数 `lengths` 指定的是每个子集的长度,因此训练集和测试集的样本数量分别为 1800 和 200。
如果 `all_dataset` 是一个包含 2000 个样本的数据集,那么经过划分后,训练集包含前 1800 个样本,测试集包含后 200 个样本。如果 `all_dataset` 中的样本是随机排序的,那么划分后的训练集和测试集中的样本应该也是随机的。
阅读全文