dmlc--dgl
Hongzhi (Steve), Chen
704bcaf6dd
Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
Relational Graph Attention Networks (RGAT)
This is an adaptation of RGCN where graph convolution is replaced with graph attention.
Dependencies
- torchmetrics
Install as follows:
pip install torchmetrics
How to Run
Run with the following for node classification on ogbn-mag dataset
python train.py
Summary
- ogbn-mag (test acc.): ~0.3647