Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks

About

We present Wiki-CS, a novel dataset derived from Wikipedia for benchmarking Graph Neural Networks. The dataset consists of nodes corresponding to Computer Science articles, with edges based on hyperlinks and 10 classes representing different branches of the field. We use the dataset to evaluate semi-supervised node classification and single-relation link prediction models. Our experiments show that these methods perform well on a new domain, with structural properties different from earlier benchmarks. The dataset is publicly available, along with the implementation of the data pipeline and the benchmark experiments, at https://github.com/pmernyei/wiki-cs-dataset .

P\'eter Mernyei, C\u{a}t\u{a}lina Cangea• 2020

Related benchmarks

TaskDatasetResultRank
Node ClassificationCora
Accuracy87.87
885
Node ClassificationCiteseer
Accuracy76.53
804
Node ClassificationPubmed
Accuracy89.4
742
Node ClassificationwikiCS
Accuracy69.83
198
Node ClassificationOgbn-arxiv
Accuracy66.38
191
Node ClassificationUS County
Accuracy84.14
14
Node ClassificationRice31
Accuracy11.34
14
Node ClassificationEmail
Accuracy70.28
9
Showing 8 of 8 rows

Other info

Follow for update