(self, data, num_hops=1, max_nodes=100, dataset_name="Arxiv", transform=None, pre_transform=None)
| 146 | |
| 147 | class kHopSubgraphDataset_Arxiv(Dataset): |
| 148 | def __init__(self, data, num_hops=1, max_nodes=100, dataset_name="Arxiv", transform=None, pre_transform=None): |
| 149 | super(kHopSubgraphDataset_Arxiv, self).__init__(None, transform, pre_transform) |
| 150 | self.data = data |
| 151 | self.num_hops = num_hops |
| 152 | self.unique_classes = data.y.unique() |
| 153 | self.k_over_2 = len(self.unique_classes) // 2 |
| 154 | if dataset_name == "Citeseer": |
| 155 | self.k_over_2 = 2 |
| 156 | elif dataset_name == "Arxiv": |
| 157 | self.k_over_2 = 10 |
| 158 | self.max_nodes = max_nodes |
| 159 | self.dataset_name = dataset_name |
| 160 | self.file_path = self.dataset_name + '_index.pkl' |
| 161 | if os.path.exists(self.file_path): |
| 162 | with open(self.file_path, 'rb') as f: |
| 163 | self.valid_subgraphs = pickle.load(f) |
| 164 | else: |
| 165 | self.valid_subgraphs = self._find_valid_subgraphs() |
| 166 | |
| 167 | def _find_valid_subgraphs(self): |
| 168 | valid_subgraphs = [] |
nothing calls this directly
no test coverage detected