gammagl.datasets

gammagl.datasets.Amazon

The Amazon Computers and Amazon Photo networks from the "Pitfalls of Graph Neural Network Evaluation" paper.

gammagl.datasets.Coauthor

The Coauthor CS and Coauthor Physics networks from the "Pitfalls of Graph Neural Network Evaluation" paper.

gammagl.datasets.TUDataset

A variety of graph kernel benchmark datasets, .e.g. "IMDB-BINARY", "REDDIT-BINARY" or "PROTEINS", collected from the TU Dortmund University.

gammagl.datasets.Planetoid

The citation network datasets "Cora", "CiteSeer" and "PubMed" from the "Revisiting Semi-Supervised Learning with Graph Embeddings" paper.

gammagl.datasets.Reddit

The Reddit dataset from the "Inductive Representation Learning on Large Graphs" paper, containing Reddit posts belonging to different communities.

gammagl.datasets.IMDB

A subset of the Internet Movie Database (IMDB), as collected in the "MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding" paper.

gammagl.datasets.Entities

The relational entities networks "AIFB", "MUTAG", "BGS" and "AM" from the "Modeling Relational Data with Graph Convolutional Networks" paper.

gammagl.datasets.Flickr

The Flickr dataset from the "GraphSAINT: Graph Sampling Based Inductive Learning Method" paper, containing descriptions and common properties of images.

gammagl.datasets.HGBDataset

A variety of heterogeneous graph benchmark datasets from the "Are We Really Making Much Progress? Revisiting, Benchmarking, and Refining Heterogeneous Graph Neural Networks" paper.

gammagl.datasets.WikipediaNetwork

The Wikipedia networks introduced in the "Multi-scale Attributed Node Embedding" paper.

gammagl.datasets.WebKB

The WebKB datasets used in the "Geom-GCN: Geometric Graph Convolutional Networks" paper.

gammagl.datasets.ModelNet40

The ModelNet40 benchmark dataset used for classification from the "Dynamic Graph CNN for Learning on Point Clouds" paper, containing 12,311 meshed CAD models from 40 categories.

gammagl.datasets.DBLP

A subset of the DBLP computer science bibliography website, as collected in the "MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding" paper.

gammagl.datasets.CA_GrQc

The CA-GrQc datasets used in the "GraphGAN: Graph Representation Learning with Generative Adversarial Nets" paper.

gammagl.datasets.ZINC

The ZINC dataset from the ZINC database and the "Automatic Chemical Design Using a Data-Driven Continuous Representation of Molecules" paper, containing about 250,000 molecular graphs with up to 38 heavy atoms.

gammagl.datasets.AMiner

The heterogeneous AMiner dataset from the "metapath2vec: Scalable Representation Learning for Heterogeneous Networks" paper, consisting of nodes from type "paper", "author" and "venue".

gammagl.datasets.PolBlogs

The Political Blogs dataset from the "The Political Blogosphere and the 2004 US Election: Divided they Blog" paper.

gammagl.datasets.WikiCS

The semi-supervised Wikipedia-based dataset from the "Wiki-CS: A Wikipedia-Based Benchmark for Graph Neural Networks" paper, containing 11,701 nodes, 216,123 edges, 10 classes and 20 different training splits.