README.md 812 Bytes
Newer Older
Shaked Brody's avatar
Shaked Brody committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
Graph Attention Networks v2 (GATv2)
============

- Paper link: [How Attentive are Graph Attention Networks?](https://arxiv.org/pdf/2105.14491.pdf)
- Author's code repo: [https://github.com/tech-srl/how_attentive_are_gats](https://github.com/tech-srl/how_attentive_are_gats).
- Annotated implemetnation: [https://nn.labml.ai/graphs/gatv2/index.html]

Dependencies
------------
- torch
- requests
- sklearn

How to run
----------

Run with following:

```bash
python3 train.py --dataset=cora
```

```bash
python3 train.py --dataset=citeseer
```

```bash
python3 train.py --dataset=pubmed
```

Results
-------

| Dataset  | Test Accuracy |
| -------- | ------------- |
| Cora     |  82.10        |
| Citeseer |  70.00        |
| Pubmed   |  77.2         |

* All the accuracy numbers are obtained after 200 epochs.