Please install pakeages by
pip install -r requirements.txt
Cora
python main.py --dataset cora --runs 100 --epochs 600 --batch_size 2200 --dropout 0 --hidden 300 --hidden_z 300 --early_stopping 20 --lr 0.007 --weight_decay 0.0004 --alpha 2 --beta 3 --tau 0.5 --order 4
CiteSeer
python main.py --dataset citeseer --epochs 500 --batch_size 3000 --lr 0.01 --weight_decay 0.0005 --dropout 0.8 --hidden 1500 --hidden_z 500 --early_stopping 10 --alpha 1 --beta 2 --tau 0.5 --order 4
PubMed
python main.py --dataset pubmed --runs 100 --epochs 400 --batch_size 2500 --lr 0.01 --weight_decay 0.0005 --dropout 0.2 --hidden 400 --hidden_z 400 --early_stopping 10 --alpha 10 --beta 1 --tau 0.5 --order 4
Amazon Computers
python main.py --dataset computers --runs 100 --epochs 300 --lr 0.005 --weight_decay 0.0005 --dropout 0.4 --hidden 400 --hidden_z 300 --early_stopping 10 --alpha 30 --beta 3 --tau 4 --order 6
Amazon Photo
python main.py --dataset photo --runs 100 --epochs 200 --lr 0.005 --weight_decay 0.0005 --dropout 0.5 --hidden 200 --hidden_z 200 --early_stopping 10 --alpha 25 --beta 3 --tau 4 --order 5
Coauthor CS
python main.py --dataset cs --runs 10 --epochs 100 --lr 0.01 --weight_decay 0.005 --dropout 0.8 --hidden 2000 --hidden_z 600 --early_stopping 10 --alpha 10 --beta 1 --tau 1.2 --order 2
model | Cora | CiteSeer | PubMed | Amazon Computers | Amazon Photo | Coauthor CS |
---|---|---|---|---|---|---|
NE-WNA | 82.8% | 74.2% | 82.5% | 84.7% | 93.2% | 92.5% |