dmlc--dgl
a1f7498240
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>
26 行
427 B
Markdown
26 行
427 B
Markdown
Relational Graph Attention Networks (RGAT)
|
|
==============
|
|
This is an adaptation of RGCN where graph convolution is replaced with graph attention.
|
|
|
|
Dependencies
|
|
------------
|
|
- torchmetrics 0.11.4
|
|
|
|
Install as follows:
|
|
```bash
|
|
pip install torchmetrics==0.11.4
|
|
```
|
|
|
|
How to Run
|
|
-------
|
|
|
|
Run with the following for node classification on ogbn-mag dataset
|
|
```bash
|
|
python train.py
|
|
```
|
|
|
|
|
|
Summary
|
|
-------
|
|
* ogbn-mag (test acc.): ~0.3647
|