柴火饭
2022-04-05 15:31
采纳率: 33.3%
浏览 70

如何把cifar10的一万个测试数据集划分成 训练、验证、测试数据集

敲完下面代码我就不知道该怎么办了,想知道如何把这一万个数据进行划分?

import torch
import torchvision
import torchvision.transforms as transforms
from sklearn.model_selection import train_test_split

transform = transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])

Data = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform)

2条回答 默认 最新

相关推荐 更多相似问题