(self, root_path, win_size, step=1, flag="train")
| 545 | |
| 546 | class SWATSegLoader(Dataset): |
| 547 | def __init__(self, root_path, win_size, step=1, flag="train"): |
| 548 | self.flag = flag |
| 549 | self.step = step |
| 550 | self.win_size = win_size |
| 551 | self.scaler = StandardScaler() |
| 552 | |
| 553 | train_data = pd.read_csv(os.path.join(root_path, 'swat_train2.csv')) |
| 554 | test_data = pd.read_csv(os.path.join(root_path, 'swat2.csv')) |
| 555 | labels = test_data.values[:, -1:] |
| 556 | train_data = train_data.values[:, :-1] |
| 557 | test_data = test_data.values[:, :-1] |
| 558 | |
| 559 | self.scaler.fit(train_data) |
| 560 | train_data = self.scaler.transform(train_data) |
| 561 | test_data = self.scaler.transform(test_data) |
| 562 | self.train = train_data |
| 563 | self.test = test_data |
| 564 | self.val = test_data |
| 565 | self.test_labels = labels |
| 566 | print("test:", self.test.shape) |
| 567 | print("train:", self.train.shape) |
| 568 | |
| 569 | def __len__(self): |
| 570 | """ |
nothing calls this directly
no test coverage detected