- pytorch 1.1+
Run with following (available dataset: "cora", "citeseer", "pubmed")
python3 citation.py --dataset cora --gpu 0
- Cora: ~0.816
- Pubmed: ~0.763
- please refer to model_zoo/geometric.
Run with following (available dataset: "cora", "citeseer", "pubmed")
python3 citation.py --dataset cora --gpu 0