| 414 | |
| 415 | |
| 416 | class IndicesDataset: |
| 417 | def __init__(self, dataset, indices=None): |
| 418 | self.dataset = dataset |
| 419 | self.indices = range(len(dataset)) if (indices is None) else indices |
| 420 | |
| 421 | def __getitem__(self, i): |
| 422 | return self.dataset[self.indices[i]] |
| 423 | |
| 424 | def __len__(self): |
| 425 | return len(self.indices) |
| 426 | |
| 427 | |
| 428 | class TriggerTrans: |