(dataset_name)
| 32 | |
| 33 | |
| 34 | def load_dataset(dataset_name): |
| 35 | dataset = build_dataset_from_name(dataset_name) |
| 36 | if dataset_name == "ogbn-arxiv": |
| 37 | graph = dataset[0] |
| 38 | feat = graph.x |
| 39 | feat = scale_feats(feat) |
| 40 | graph.x = feat |
| 41 | |
| 42 | else: |
| 43 | graph = dataset[0] |
| 44 | graph.add_remaining_self_loops() |
| 45 | |
| 46 | num_features = graph.x.shape[1] |
| 47 | num_classes = dataset.num_classes |
| 48 | return graph, (num_features, num_classes) |
| 49 | |
| 50 | def load_inductive_dataset(dataset_name): |
| 51 | dataset = build_dataset_from_name(dataset_name) |
no test coverage detected