- MXNet 1.5+
Run with following (available dataset: "cora", "citeseer", "pubmed")
python3 citation.py --dataset cora --gpu 0
- Cora: ~0.814
- Pubmed: ~0.748
Run with following (available dataset: "cora", "citeseer", "pubmed")
python3 citation.py --dataset cora --gpu 0