Update util.py

import torch

import os

import glob

import torch.utils.data as data

class KpiReader(data.Dataset):

    def __init__(self, path, size):

    def __init__(self, path):

        super(KpiReader, self).__init__()

        self.path = path

        self.length = size

        self.length = len(glob.glob(self.path+'/*.seq'))

        data = []

        for i in range(self.length):

            item = torch.load(self.path+'/%d.seq' % (i+1))

            data.append(item)

        self.data = data

    def __getitem__(self, index):

        kpi_ts, kpi_label, kpi_value = self.data[index]['ts'],self.data[index]['label'],self.data[index]['value']

        return kpi_ts, kpi_label, kpi_value

    def __len__(self):

        return self.length

    def __getitem__(self, idx):

        item = torch.load(self.path+'/%d.seq' % (idx+1))

        return item['ts'], item['label'], item['value']