【PyTorch】数据的读取和操作(Dataset, DataLoader)

前言

Pytorch 中比较重要的是对数据的处理，其中，进行数据读取的一般有三个类：

• Dataset
• DataLoader
• DataLoaderIter

其中，这三是一个依次封装的关系：“Dataset被封装进DataLoader，DataLoader再被封装进DataLoaderIter”

Dataset

class torch.utils.data.Dataset 

表示Dataset的抽象类，所有其数据集都应该继承这个类。所有子类必须继承和实现

• len()
• getitem()

两个成员函数，前者提供了数据集的大小，后者支持整数索引，范围从0到len(self)。

import torch
from torch.utils.data import Dataset
import pandas as pd

# 定义自己的类
class MyDataset(Dataset):
    
    # 初始化
    def __init__(self, file_name):
        # 读入数据
        self.data