(self)
| 35 | super(MyGraphClassificationDataset, self).__init__(path) |
| 36 | |
| 37 | def process(self): |
| 38 | graphs = [] |
| 39 | for i in range(200): |
| 40 | edges = torch.randint(0, 1000, (2, 30)) |
| 41 | label = torch.randint(0, 7, (1,)) |
| 42 | graphs.append(Graph(edge_index=edges, y=label)) |
| 43 | torch.save(graphs, self.path) |
| 44 | return graphs |
| 45 | |
| 46 | |
| 47 | def test_customized_dataset(): |