PyTorch加载数据集

最新推荐文章于 2025-02-13 11:28:31 发布

心️升明月

最新推荐文章于 2025-02-13 11:28:31 发布

阅读量1.3k

点赞数 3

分类专栏：机器学习文章标签： pytorch 机器学习 DataLoader DataSet mini_batch

本文链接：https://blog.csdn.net/weixin_43821559/article/details/123328242

版权

一、实现过程

1、准备数据

与PyTorch实现多维度特征输入的逻辑回归的方法不同的是：本文使用DataLoader方法，并继承DataSet抽象类，可实现对数据集进行mini_batch梯度下降优化。代码如下：

import torch
import numpy as np
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[:,[