This is the official pytorch implementation of Collaborative Graph Neural Networks for Attributed Network Embedding.
The required packages can be installed by running pip install -r requirements.txt
.
The datasets used in our paper can be automatically downlowad.
Test on node classification task:
python main_classify.py --dataset "PubMed"
Test on link prediction task:
python main_lp.py --dataset "PubMed"