README.md 412 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
7
Dependencies
------------
- torchmetrics
8

9
10
11
12
Install as follows:
```bash
pip install torchmetrics
```
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