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