刘二大人《PyTorch深度学习实践》——第8讲加载数据集

最新推荐文章于 2024-05-28 19:45:06 发布

javaft

最新推荐文章于 2024-05-28 19:45:06 发布

阅读量1.7k

点赞数

文章标签： python pytorch

本文链接：https://blog.csdn.net/javaft/article/details/124043887

版权

刘二大人《PyTorch深度学习实践》——第8讲加载数据集（代码详解）

import numpy as np
import torch
from torch.utils.data import Dataset, DataLoader
class DiabetesDataset(Dataset):  # 定义一个糖尿病人数据集类，继承自Dataset
    def __init__(self, filepath):
        xy = np.loadtxt(filepath, delimiter=',', dtype=np.float32)  # 加载训练数据集
        self.len = xy.shape[0]  # shape本身是一个二元组（x,y）对应数据集的行数和列数，这里[0]我们取行数,即样本数
        self.x_data = torch.from_numpy(xy[:, :-1])  # 取前八列   第一个‘：’是指读取所有行，第二个‘：’是指从第一列开始，最后一列不要
        self.y_data = torch.</