(self, root_path, flag='pred', size=None,
features='S', data_path='ETTh1.csv',
target='OT', scale=True, inverse=False, timeenc=0, freq='15min', cols=None)
| 292 | |
| 293 | class Dataset_Pred(Dataset): |
| 294 | def __init__(self, root_path, flag='pred', size=None, |
| 295 | features='S', data_path='ETTh1.csv', |
| 296 | target='OT', scale=True, inverse=False, timeenc=0, freq='15min', cols=None): |
| 297 | # size [seq_len, label_len, pred_len] |
| 298 | # info |
| 299 | if size == None: |
| 300 | self.seq_len = 24 * 4 * 4 |
| 301 | self.label_len = 24 * 4 |
| 302 | self.pred_len = 24 * 4 |
| 303 | else: |
| 304 | self.seq_len = size[0] |
| 305 | self.label_len = size[1] |
| 306 | self.pred_len = size[2] |
| 307 | # init |
| 308 | assert flag in ['pred'] |
| 309 | |
| 310 | self.features = features |
| 311 | self.target = target |
| 312 | self.scale = scale |
| 313 | self.inverse = inverse |
| 314 | self.timeenc = timeenc |
| 315 | self.freq = freq |
| 316 | self.cols = cols |
| 317 | self.root_path = root_path |
| 318 | self.data_path = data_path |
| 319 | self.__read_data__() |
| 320 | |
| 321 | def __read_data__(self): |
| 322 | self.scaler = StandardScaler() |
nothing calls this directly
no test coverage detected