/OLF-GCL

Primary LanguagePython

OLF-GCL

环境:gpu-3090 python=3.9.18 pytorch=1.11.0 cuda=11.3

安装步骤:

1.conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

2.conda install -c dglteam dgl-cuda11.3==0.9.1post1

3.pip install scikit_learn

run:

citeseer: python train.py --dataset=citeseer --alpha=10 --dc=0.05 --tree_nums=10 --n-layers=1 --patience=40 --wd2=0.01

pubmed:python train.py --dataset=pubmed --alpha=10 --dc=0.05 --tree_nums=80 --n-classifier-epochs=1500 --beta=100

photo: python train.py --dataset=photo --alpha=15 --dc=1 --tree_nums=10 --n-classifier-epochs=900

computer: python train.py --dataset=comp --alpha=15 --dc=1 --tree_nums=10 --n-classifier-epochs=2000