(self, idx)
| 99 | return self.subgraphs[idx] |
| 100 | |
| 101 | def __getitem__(self, idx): |
| 102 | return self.get(idx) |
| 103 | |
| 104 | # class kHopSubgraphDataset(Dataset): |
| 105 | # def __init__(self, data, num_hops=1, max_nodes=100, dataset_name="Cora", min_classes=None, transform=None, pre_transform=None): |