torch.utils.data.Dataset用法

利用继承自Dataset的类，可以访问训练所需的数据
比如一下数据：保存为csv文件

from torch.utils.data import Dataset
import pandas as pd #这个包用来读取CSV数据
class mydataset(Dataset):def __init__(self,csv_file): #self参数必须，其他参数及其形式随程序需要而不同，比如(self,*inputs)self.csv_data=pd.read_csv(csv_file)def __len__(self):return len(self.csv_data)def __getitem__(self,idx):data=self.csv_data.values[idx]return data
data=mydataset('/home/yls/Documents/test.csv')
print(data[3])
print(len(data))

输出结果如下：
[‘OpenSuse’ ‘stable’ ‘OpenSuse Repository’
‘zypper\xa0in\xa0python3-pandas’]
6

torch.utils.data.Dataset用法相关推荐

阅读源码-理解torch.utils.data、torch.utils.data.Dataset、torch.utils.data.DataLoader的工作方式
文章目录目标 Dataset DataLoader 应用 Dataset DataLoader 测试知识点 Python splitlines()方法 python filter()函数暂时先写 ...
5 torch.utils.data (Dataset,TensorDataset,DataLoader)
文章目录一.DataLoader(数据预处理) 1.DataLoader :(构建可迭代的数据装载器) 2.输出:DataLoader 的输出包含:数据和标签二.TensorDataset(数据预 ...
【PyTorch】torch.utils.data.Dataset 介绍与实战
文章目录一.前言二.torch.utils.data.Dataset 是什么 1. 干什么用的? 2. 长什么样子? 三.通过继承 torch.utils.data.Dataset 定义自己的数据 ...
pytorch torch.utils.data.Dataset
应用 from torch.utils.data import DataLoader, Dataset import torchclass TensorDataset(Dataset):# Tenso ...
PyTorch 1.0 中文文档：torch.utils.data
译者:BXuan694 class torch.utils.data.Dataset 表示数据集的抽象类. 所有用到的数据集都必须是其子类.这些子类都必须重写以下方法:__len__:定义了数据集的规 ...
pytorch源码解析2——数据处理torch.utils.data
迭代器理解 Python 的迭代器是解读 PyTorch 中 torch.utils.data 模块的关键. 在 Dataset, Sampler 和 DataLoader 这三个类中都会用到 py ...
PyTorch—torch.utils.data.DataLoader 数据加载类
文章目录 DataLoader(object)类: _DataLoaderIter(object)类 __next__函数 pin_memory_batch() _get_batch函数 _proce ...
Pytorch 中的数据类型 torch.utils.data.DataLoader 参数详解
DataLoader是PyTorch中的一种数据类型,它定义了如何读取数据方式.详情也可参考本博主的另一篇关于torch.utils.data.DataLoader(https://blog.csdn ...
PyTorch 源码解读之 torch.utils.data：解析数据处理全流程
目录 0 前言 1 Dataset 1.1 Map-style dataset 1.2 Iterable-style dataset 1.3 其他 dataset 2 Sampler 3 DataLo ...
torch.utils.data.WeightedRandomSampler样本不均衡情况下带权重随机采样
关于WeightedRandomSampler的用法csdn上有一些很棒的博客.本文参考博客Pytorch样本比例不均衡时采用WeightedRandomSampler进行采样的代码对Weighted ...

torch.utils.data.Dataset用法

torch.utils.data.Dataset用法相关推荐

最新文章

热门文章