Skip to content

Latest commit

 

History

History
31 lines (18 loc) · 836 Bytes

README.md

File metadata and controls

31 lines (18 loc) · 836 Bytes

Introduction

This repository is the implementation of CIKM'23 paper: Semantic-aware Node Synthesis for Imbalanced Heterogeneous Information Networks.


SNS

Requirements

All experiments are implemented in Python 3.9 with Pytorch 1.11.0.

To install requirements:

pip install -r requirements.txt

Data

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.

Model training

To train the model and execute the SNS under im_ratio=0.1:

$ cd ./scr
$ bash ./run.sh