本文共 3702 字，大约阅读时间需要 12 分钟。

PyTorch ���������������������������������

������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������

���������������Dataset Class������������������

���PyTorch������������������������������������torch.utils.data.Dataset���������������������������������������������������������������������������������������������������������������������������������

������������Dataset���������������������������������������������������������������������������������������������

import pandas as pdfrom torch.utils.data import Datasetclass MyDataset(Dataset):    def __init__(self):        self dataSet = pd.read_csv("./data/wendu_8_4_9_2.csv").values        def __getitem__(self, index):        return selfDataSetToTensor(selfDataSet[index])        def __len__(self):        return len(selfDataSet)        @staticmethod    def toTensor(data):        return torch.from_numpy(data)

������������������������������CSV���������������������PyTorch������������������MyDSataset������������������������������������������������������������������������toTensor()���������������PyTorch���������������

������������������DataLoader������������������������������������

���������������������������������������������������������������������������������������������������������������������PyTorch������DataLoader���������������������������������������������

• ������������������������������������������������������������������
• ���������������������������������������������������������������
• ���������������������������������������������������������������������
• ������������������������������������������������������������������������������������������������

DataLoader������������������������

from torch.utils.data import DataLoaderimport pandas as pdimport torchdataSet =MyDataset() # ���������������������������# ������������������������dataLoader = DataLoader(    dataSet=dataSet,    batch_size=32,    shuffle=True,    num_workers=2)for batch in dataLoader:    print(batch)    print('*' * 50)

DataLoader���������������������

���������DataLoader��������������������������������� batches������������������������batches���������������������������������������������������������������������������������

���������������������������

���������DataLoader���������������������������������

������������������DataLoader���������������������������������������������������������������������������������������������������