该专栏为热销专栏榜 第8名
超级会员免费看
本文介绍了如何在Pytorch中构建自定义数据集。通过继承Pytorch的Dataset类，可以方便地处理和加载个人数据作为训练和测试集。文章详细讲解了定义数据集和枚举获取数据的步骤。
Build my Dataset in Pytorch.
在使用Pytorch进行项目时，有时候需要读入自己的数据作为训练集和测试集，并按照自己指定的方式和格式处理。
Pytorch定义了Dataset类，在实际使用中可以通过继承Dataset类来构建数据集：
from torch.utils.data import Dataset, DataLoader
class myData(Dataset):
def 
订阅专栏 解锁全文
扫一扫
评论 8
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
