(self, root_path, win_size, step=1, flag="train")
| 373 | |
| 374 | class PSMSegLoader(Dataset): |
| 375 | def __init__(self, root_path, win_size, step=1, flag="train"): |
| 376 | self.flag = flag |
| 377 | self.step = step |
| 378 | self.win_size = win_size |
| 379 | self.scaler = StandardScaler() |
| 380 | data = pd.read_csv(os.path.join(root_path, 'train.csv')) |
| 381 | data = data.values[:, 1:] |
| 382 | data = np.nan_to_num(data) |
| 383 | self.scaler.fit(data) |
| 384 | data = self.scaler.transform(data) |
| 385 | test_data = pd.read_csv(os.path.join(root_path, 'test.csv')) |
| 386 | test_data = test_data.values[:, 1:] |
| 387 | test_data = np.nan_to_num(test_data) |
| 388 | self.test = self.scaler.transform(test_data) |
| 389 | self.train = data |
| 390 | self.val = self.test |
| 391 | self.test_labels = pd.read_csv(os.path.join(root_path, 'test_label.csv')).values[:, 1:] |
| 392 | print("test:", self.test.shape) |
| 393 | print("train:", self.train.shape) |
| 394 | |
| 395 | def __len__(self): |
| 396 | if self.flag == "train": |
nothing calls this directly
no test coverage detected