dmlc--dgl
Chang Liu
98792a8ae3
* RGAT refactor * File rename * Address comments Co-authored-by: Mufei Li <mufeili1996@gmail.com>
412 B
412 B
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