/SR-HGN

This repo is for source code of Expert Systems with Applications paper "SR-HGN: Semantic-and Relation-Aware Heterogeneous Graph Neural Network".

Primary LanguagePython

SR-HGN

This repo is for source code of Expert Systems with Applications paper "SR-HGN: Semantic-and Relation-Aware Heterogeneous Graph Neural Network". paper

Environment Settings

  • python==3.8.0
  • scipy==1.6.2
  • torch==1.11.0
  • scikit-learn==1.0.2
  • torch_geometric==2.0.4
  • dgl==0.6.1

Dataset

We utilize three benchmark datasets in the paper to perform node classification and node clustering. We provide ACM, DBLP, and IMDB in GoogleDrive.

  • ACM
  • DBLP
  • IMDB

You can create the "data" folder in the root directory, then put the datasets in. Like "/SRHGN/data/acm/...".

How to run

For example, if you want to run SR-HGN on ACM dataset, execute

python main.py --dataset acm

Citation

@article{wang2023sr,
  title={SR-HGN: Semantic-and Relation-Aware Heterogeneous Graph Neural Network},
  author={Wang, Zehong and Yu, Donghua and Li, Qi and Shen, Shigen and Yao, Shuang},
  journal={Expert Systems with Applications},
  volume={224},
  pages={119982},
  year={2023},
  publisher={Elsevier}
}

Contact

If you have any questions, don't hesitate to contact me (zwang43@nd.edu, zehongwang0414@gmail.com)!