data_utils.py

import numpy as np
import torch
from torch.utils.data import Dataset



class LPDataset(Dataset):#Dataset是数据集位置，就是下面的path

    def __init__(self, data, input_n, output_n):
        super(LPDataset, self).__init__()
        self.data = data
        self.data = torch.from_numpy(self.data)
        self.data = self.data.reshape((-1,self.data.shape[1],7))
        self.input_n = input_n
        self.output_n = output_n
        self.num = self.data.shape[0] - (input_n + output_n)
    def __len__(self):
        return self.num
        
    def __getitem__(self, item):
        a = item
        b = item + self.input_n
        c = item + self.input_n + self.output_n
        return self.data[a: b], self.data[b : c]