一、制作自己的数据集 源代码 import torch from torch.utils.data import Dataset import pandas as pd import numpy as np # txt文件内容 路径 \t 类别 \t 长度 \n txt_path = 'G:/stock/path.txt' class SocktData(Dataset): dataset = [] # 【data,label】形式初始化 def __init__(self,txt_path): fh = open(txt_path) f