(self, root_path, flag='train', size=None,
features='S', data_path='ETTh1.csv',
target='OT', scale=True, timeenc=0, freq='h')
| 191 | |
| 192 | class Dataset_Custom(Dataset): |
| 193 | def __init__(self, root_path, flag='train', size=None, |
| 194 | features='S', data_path='ETTh1.csv', |
| 195 | target='OT', scale=True, timeenc=0, freq='h'): |
| 196 | # size [seq_len, label_len, pred_len] |
| 197 | # info |
| 198 | if size == None: |
| 199 | self.seq_len = 24 * 4 * 4 |
| 200 | self.label_len = 24 * 4 |
| 201 | self.pred_len = 24 * 4 |
| 202 | else: |
| 203 | self.seq_len = size[0] |
| 204 | self.label_len = size[1] |
| 205 | self.pred_len = size[2] |
| 206 | # init |
| 207 | assert flag in ['train', 'test', 'val'] |
| 208 | type_map = {'train': 0, 'val': 1, 'test': 2} |
| 209 | self.set_type = type_map[flag] |
| 210 | |
| 211 | self.features = features |
| 212 | self.target = target |
| 213 | self.scale = scale |
| 214 | self.timeenc = timeenc |
| 215 | self.freq = freq |
| 216 | |
| 217 | self.root_path = root_path |
| 218 | self.data_path = data_path |
| 219 | self.__read_data__() |
| 220 | |
| 221 | def __read_data__(self): |
| 222 | self.scaler = StandardScaler() |
nothing calls this directly
no test coverage detected