L08_Dataset和DataLoader

最新推荐文章于 2024-07-23 16:23:28 发布

槿年的槿年

最新推荐文章于 2024-07-23 16:23:28 发布

阅读量107

点赞数

分类专栏：研究生之深度学习文章标签： python 深度学习 numpy

本文链接：https://blog.csdn.net/qq_42753878/article/details/128454311

版权

研究生之深度学习专栏收录该内容

12 篇文章 0 订阅

订阅专栏

这节主要是学习如何使用已有数据集来使用数据加载器

import numpy as np
import torch
from torch.utils.data import Dataset, DataLoader

class DiabetesDataset(Dataset):
    def __init__(self,filepath):
        xy = np.loadtxt(filepath,delimiter=',',dtype=np.float32)
        self.len = xy.shape[0]
        self.x_data = torch.from_numpy(xy[:,:-1])
        self.y_data = torch.from_numpy(xy[:,-1:])
        
    def __getitem__(self,index):
        return self.x_data[index], self.y_data[index]
    
    def __len__(self):
        return self.len
    
dataset = DiabetesDataset('./data/diabetes.csv.gz')
train_loader = DataLoader(dataset=dataset,batch_size=32,shuffle=True, num_workers=2)