自己需要完成一个基于时序信号数据的自编码器,但是在读取数据时遇到了一些困难。数据集(无标签)由csv文件构成,每个数据样本为一个csv文件,csv文件中的数据为4列512行。
博客园同步: pytorch读取自己的csv数据集
python代码
// 重写dataset类
import os
import torch
import pandas as pd
from torch.utils.data import Dataset
from torch.utils.data import DataLoader
class myDataset(Dataset):
def __init__(self, data_dir):
"""
data_dir: 数据文件路径
""""
#