arXiv · 2007.02901
Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks
Abstract
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 .
Explore related subjects
Keep this discovery
Péter Mernyei, Cătălina Cangea. 2020-07-06. Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks. https://arxiv.org/abs/2007.02901
Cite the original work for its findings. Save a collection to share your selection of sources.