torch学习 自用
1. Dataset和Dataloader
Dataset定义数据集的结构:需要实现__init__() __len__() __getitem__()这三个函数
dataloader定义如何读取数据:怎么批量加载、是否打乱、用几个进程批量读取
这里面df.iloc[],逗号前面是取行,逗号后面是取列
数据先转换成numpy类型,再转换成张量类型
「df是一个pandas.DataFrame对象」
2. 推理的代码
1. Dataset和Dataloader
Dataset定义数据集的结构:需要实现__init__() __len__() __getitem__()这三个函数
dataloader定义如何读取数据:怎么批量加载、是否打乱、用几个进程批量读取
这里面df.iloc[],逗号前面是取行,逗号后面是取列
数据先转换成numpy类型,再转换成张量类型
「df是一个pandas.DataFrame对象」
2. 推理的代码