README.md 427 Bytes
Newer Older
1
Relational Graph Attention Networks (RGAT)
2
==============
3
This is an adaptation of RGCN where graph convolution is replaced with graph attention.
4

5
6
Dependencies
------------
7
- torchmetrics 0.11.4
8

9
10
Install as follows:
```bash
11
pip install torchmetrics==0.11.4
12
```
13

14
15
How to Run
-------
16

17
Run with the following for node classification on ogbn-mag dataset
18
```bash
19
python train.py
20
21
```

22
23
24
25

Summary
-------
* ogbn-mag (test acc.): ~0.3647