()
| 42 | |
| 43 | |
| 44 | def init_dataset(): |
| 45 | dataset = Cora_GRBDataset() |
| 46 | graph = copy.deepcopy(dataset.get(0)) |
| 47 | device = "cuda:0" if torch.cuda.is_available() else "cpu" |
| 48 | device_ids = [0] |
| 49 | graph.to(device) |
| 50 | test_mask = graph.test_mask |
| 51 | set_random_seed(40) |
| 52 | return graph, dataset, test_mask, device, device_ids |
| 53 | |
| 54 | |
| 55 | def init_surrogate_model(graph, dataset, test_mask, device, device_ids): |
no test coverage detected