SHGNN: Structure-Aware Heterogeneous Graph Neural Network
The source code and dataset of the paper: SHGNN: Structure-Aware Heterogeneous Graph Neural Network.
Requirements
The framework is implemented using python3 with dependencies specified in requirements.txt.
git clone https://github.com/Wentao-Xu/SHGNN.git
cd SHGNN
conda create -n shgnn python=3.8
conda activate shgnn
pip install -r requirements.txt
Dataset preparation
source prepare_data.sh
tar -zxvf data.tar.gz
mkdir checkpoint
Running the code
# IMDB dataset
python run_IMDB.py
# DBLP dataset
python run_DBLP.py
# ACM dataset
python run_ACM.py