(self, occ, prc, lb, pt, device, adj)
| 86 | |
| 87 | class CreateDataset(Dataset): |
| 88 | def __init__(self, occ, prc, lb, pt, device, adj): # adj |
| 89 | occ, label = create_rnn_data(occ, lb, pt) |
| 90 | prc, _ = create_rnn_data(prc, lb, pt) |
| 91 | self.occ = torch.Tensor(occ) |
| 92 | self.prc = torch.Tensor(prc) |
| 93 | self.label = torch.Tensor(label) |
| 94 | self.device = device |
| 95 | |
| 96 | def __len__(self): |
| 97 | return len(self.occ) |
nothing calls this directly
no test coverage detected