Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
README.md 339 Bytes
Newer Older
rusty1s's avatar
small benchmark script  
rusty1s committed
1
2
3 
# Benchmark on Small-scale Graphs

```
rusty1s's avatar
adjust small benchmark script  
rusty1s committed
4
5
6
7
8
9 
python main.py model=gcn dataset=cora root=/tmp/datasets device=0
```

You can choose between the following models and datasets:

* **Models:** `gcn`, `gat`, `appnp`, `gcn2`
rusty1s's avatar
adjust dataset names  
rusty1s committed
10 
* **Datasets:** `cora`, `citesser`, `pubmed`, `coauthor-cs`, `coauthor-physics`, `amazon-computers`, `amazon-photo`, `wikics`