README.md 818 Bytes
Newer Older
Shaked Brody's avatar
Shaked Brody committed
1
2
3
4
5
6
7
8
9
10
11
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
12
- scikit-learn
Shaked Brody's avatar
Shaked Brody committed
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

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         |

40
* All the accuracy numbers are obtained after 200 epochs.