This repository is the implementation of CIKM'23 paper: Semantic-aware Node Synthesis for Imbalanced Heterogeneous Information Networks.
All experiments are implemented in Python 3.9 with Pytorch 1.11.0.
To install requirements:
pip install -r requirements.txt
The datasets used in our experiments are download from the repository of SHGP (github).Please download the dataset and place it in the folder ./data
. The code of dataset processing for imbalanced setting is provided in ./scr/load_data.py
, and use --im_ratio
to change the imbalance scenario.
To train the model and execute the SNS under im_ratio=0.1
:
$ cd ./scr
$ bash ./run.sh