wehub-resource-sync
39 lines
671 B
Markdown
39 lines
671 B
Markdown
Deep Graph Infomax (DGI)
|
|
========================
|
|
|
|
- Paper link: [https://arxiv.org/abs/1809.10341](https://arxiv.org/abs/1809.10341)
|
|
- Author's code repo (in Pytorch):
|
|
[https://github.com/PetarV-/DGI](https://github.com/PetarV-/DGI)
|
|
|
|
Dependencies
|
|
------------
|
|
- PyTorch 0.4.1+
|
|
- requests
|
|
|
|
```bash
|
|
pip install torch requests
|
|
```
|
|
|
|
How to run
|
|
----------
|
|
|
|
Run with following:
|
|
|
|
```bash
|
|
python3 train.py --dataset=cora --gpu=0 --self-loop
|
|
```
|
|
|
|
```bash
|
|
python3 train.py --dataset=citeseer --gpu=0
|
|
```
|
|
|
|
```bash
|
|
python3 train.py --dataset=pubmed --gpu=0
|
|
```
|
|
|
|
Results
|
|
-------
|
|
* cora: ~81.6 (81.2-82.1) (paper: 82.3)
|
|
* citeseer: ~69.4 (paper: 71.8)
|
|
* pubmed: ~76.1 (paper: 76.8)
|