(dataset, data_path=datadir, normalization="AugNormAdj", porting_to_torch=True, task_type = "full")
| 160 | |
| 161 | |
| 162 | def data_loader(dataset, data_path=datadir, normalization="AugNormAdj", porting_to_torch=True, task_type = "full"): |
| 163 | if dataset == "reddit": |
| 164 | return load_reddit_data(normalization, porting_to_torch, data_path) |
| 165 | else: |
| 166 | (adj, |
| 167 | features, |
| 168 | labels, |
| 169 | idx_train, |
| 170 | idx_val, |
| 171 | idx_test, |
| 172 | degree, |
| 173 | learning_type) = load_citation(dataset, normalization, porting_to_torch, data_path, task_type) |
| 174 | train_adj = adj |
| 175 | train_features = features |
| 176 | return adj, train_adj, features, train_features, labels, idx_train, idx_val, idx_test, degree, learning_type |
| 177 |
no test coverage detected