jjfraaa
/
AutoGL
Not watched
1
0
Fork 0
Code
Releases 3 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 Commits
2 Branches
1.9 MB
143 Download
 
 
 
 
 
Tree: b9efe52562
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b9efe52562'
${ noResults }
AutoGL/autogl/datasets
Frozenmad 7302e85847
Merge branch 'benchmark' into fe_test 		5 years ago
..
README.md Update README in datasets 5 years ago
__init__.py Temporarily override the get method for the PyG datasets 5 years ago
gatne.py AutoGL first commit 5 years ago
gtn_data.py AutoGL first commit 5 years ago
han_data.py AutoGL first commit 5 years ago
matlab_matrix.py AutoGL first commit 5 years ago
modelnet.py Temporarily override the get method for the PyG datasets 5 years ago
ogb.py Temporarily override the get method for the PyG datasets 5 years ago
pyg.py Temporarily override the get method for the PyG datasets 5 years ago
utils.py Merge branch 'benchmark' into fe_test 5 years ago

README.md

Datasets are derived from PyG, OGB and CogDL.

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.)
Dataset PyG CogDL x y edge_index edge_attr train/val/test node train/val/test mask
Cora ✓ ✓ ✓ ✓ ✓ ✓
Citeseer ✓ ✓ ✓ ✓ ✓ ✓
Pubmed ✓ ✓ ✓ ✓ ✓ ✓
Amazon Computers ✓ ✓ ✓ ✓ ✓
Amazon Photo ✓ ✓ ✓ ✓ ✓
Coauthor CS ✓ ✓ ✓ ✓ ✓
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:
In future version, AutoGL will support the following benchmarks for different tasks:

  • unsupervised node classification: PPI, Blogcatalog, Wikipedia
  • heterogeneous node classification: DBLP, ACM, IMDB
  • link prediction: PPI, Wikipedia, Blogcatalog
  • multiplex link prediction: Amazon, YouTube, Twitter
  • link prediction datasets from OGB: ogbl-ppa, ogbl-collab, ogbl-ddi, ogbl-citation, ogbl-wikikg and ogbl-biokg.
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

图自动机器学习平台与工具包

Python SVG Markdown HTML+Django other