Browse Source

Update README in datasets

tags/v0.3.1
SwiftieH 5 years ago
parent
commit
0e2acedb68
1 changed files with 24 additions and 26 deletions
  1. +24
    -26
      autogl/datasets/README.md

+ 24
- 26
autogl/datasets/README.md View File

@@ -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.

<!--
| Dataset | PyG | CogDL | x | y | edge_index | edge_attr | train/val/test node | train/val/test mask | adj|
| ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- | ---- |
| ACM | | ✓ | ✓ | ✓ | ✓ | | ✓ | | ✓ list |
@@ -41,27 +54,12 @@ Autograph now supports the following benchmarks for different tasks:
| Amazon | | ✓ | | | | | ✓ data | | |
| Twitter | | ✓ | | | | | ✓ data | | |
| Youtube | | ✓ | | | | | ✓ data | | |
| Cora | ✓ | | ✓ | ✓ | ✓ | ✓ | | ✓ | |
| Citeseer | ✓ | | ✓ | ✓ | ✓ | ✓ | | ✓ | |
| Pubmed | ✓ | | ✓ | ✓ | ✓ | ✓ | | ✓ | |
| Reddit | ✓ | | ✓ | ✓ | ✓ | ✓ | | ✓ | |
| Mutag | ✓ | | ✓ | ✓ | ✓ | ✓ | | | |
| IMDB-B | ✓ | | | ✓ | ✓ | | | | |
| IMDB-M | ✓ | | | ✓ | ✓ | | | | |
| PROTEINS | ✓ | | ✓ | ✓ | ✓ | | | | |
| COLLAB | ✓ | | | ✓ | ✓ | | | | |
| NCI1 | ✓ | | ✓ | ✓ | ✓ | | | | |
| NCI109 | ✓ | | ✓ | ✓ | ✓ | | | | |
| Enzyme | ✓ | | ✓ | ✓ | ✓ | | | | |
| Reddit-B | ✓ | | | ✓ | ✓ | | | | |
| Reddit-Multi-5k | ✓ | | | ✓ | ✓ | | | | |
| Reddit-Multi-12k | ✓ | | | ✓ | ✓ | | | | |
| PTC-MR | ✓ | | ✓ | ✓ | ✓ | ✓ | | | |
| NCI1 | ✓ | | ✓ | ✓ | ✓ | | | | |
| NCI109 | ✓ | | ✓ | ✓ | ✓ | | | | |
| Enzyme | ✓ | | ✓ | ✓ | ✓ | | | | |







-->


Loading…
Cancel
Save