(self, idx)
| 97 | return len(self.occ) |
| 98 | |
| 99 | def __getitem__(self, idx): # occ: batch, seq, node |
| 100 | output_occ = torch.transpose(self.occ[idx, :, :], 0, 1).to(self.device) |
| 101 | output_prc = torch.transpose(self.prc[idx, :, :], 0, 1).to(self.device) |
| 102 | output_label = self.label[idx, :].to(self.device) |
| 103 | return output_occ, output_prc, output_label |
| 104 | |
| 105 | |
| 106 | class CreateFastDataset(Dataset): |
nothing calls this directly
no outgoing calls
no test coverage detected