README.md 738 字节
Newer Older
F
fengshikun01 已提交
1 2 3
# DeeperGCN: All You Need to Train Deeper GCNs

see more information in https://arxiv.org/pdf/2006.07739.pdf
F
fengshikun01 已提交
4 5 6 7 8 9 10 11 12 13 14


### Datasets

The datasets contain three citation networks: CORA, PUBMED, CITESEER. The details for these three datasets can be found in the [paper](https://arxiv.org/abs/1609.02907).

### Dependencies

- paddlepaddle>=1.6
- pgl

F
fengshikun01 已提交
15 16 17 18 19 20 21 22
### Performance

We train our models for 200 epochs and report the accuracy on the test dataset.

| Dataset | Accuracy |
| --- | --- |
| Cora | ~77% | 

F
fengshikun01 已提交
23 24 25 26 27 28 29 30 31 32 33
### How to run

For examples, use gpu to train gat on cora dataset.
```
python train.py --dataset cora --use_cuda
```

#### Hyperparameters

- dataset: The citation dataset "cora", "citeseer", "pubmed".
- use_cuda: Use gpu if assign use_cuda.