From 0e2acedb68aef6be1063e0ea4db644e24db4be50 Mon Sep 17 00:00:00 2001 From: SwiftieH Date: Tue, 22 Dec 2020 11:39:46 +0000 Subject: [PATCH] Update README in datasets --- autogl/datasets/README.md | 50 +++++++++++++++++++-------------------- 1 file changed, 24 insertions(+), 26 deletions(-) diff --git a/autogl/datasets/README.md b/autogl/datasets/README.md index 7764fd5..416bcb5 100644 --- a/autogl/datasets/README.md +++ b/autogl/datasets/README.md @@ -1,7 +1,7 @@ -Datasets are derived from CogDL +Datasets are derived from PyG, OGB and CogDL. ================= -Autograph now supports the following benchmarks for different tasks: +AutoGL now supports the following benchmarks for different tasks: - semi-supervised node classification: Cora, Citeseer, Pubmed, Amazon Computers\*, Amazon Photo\*, Coauthor CS\*, Coauthor Physics\*, Reddit （\*: using `utils.random_splits_mask_class` for splitting dataset is recommended.) @@ -16,19 +16,32 @@ Autograph now supports the following benchmarks for different tasks: | Coauthor Physics | ✓ | | ✓ | ✓ | ✓ | ✓ | | | | Reddit | ✓ | | ✓ | ✓ | ✓ | ✓ | | ✓ | + +- supervised graph classification: MUTAG, IMDB-B, IMDB-M, PROTEINS, COLLAB + +| Dataset | PyG | CogDL | x | y | edge_index | edge_attr | train/val/test node | train/val/test mask | adj| +| ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | +| Mutag | ✓ | | ✓ | ✓ | ✓ | ✓ | | | | +| IMDB-B | ✓ | | | ✓ | ✓ | | | | | +| IMDB-M | ✓ | | | ✓ | ✓ | | | | | +| PROTEINS | ✓ | | ✓ | ✓ | ✓ | | | | | +| COLLAB | ✓ | | | ✓ | ✓ | | | | | + +- node classification datasets from OGB: ogbn-products, ogbn-proteins, ogbn-arxiv, ogbn-papers100M and ogbn-mag. + +- graph classification datasets from OGB: ogbg-molhiv, ogbg-molpcba, ogbg-ppa and ogbg-code. + --- TODO: -Autograph now supports the following benchmarks for different tasks: +In future version, AutoGL will support the following benchmarks for different tasks: - unsupervised node classification: PPI, Blogcatalog, Wikipedia -- semi-supervised node classification: Cora, Citeseer, Pubmed - heterogeneous node classification: DBLP, ACM, IMDB - link prediction: PPI, Wikipedia, Blogcatalog - multiplex link prediction: Amazon, YouTube, Twitter -- unsupervised graph classification: MUTAG, IMDB-B, IMDB-M, PROTEINS, COLLAB -- supervised graph classification: MUTAG, IMDB-B, IMDB-M, PROTEINS, COLLAB - +- link prediction datasets from OGB: ogbl-ppa, ogbl-collab, ogbl-ddi, ogbl-citation, ogbl-wikikg and ogbl-biokg. +