RHGN
Source code for CIKM 2021 paper for Relation-aware Heterogeneous Graph for User Profiling
Dependencies
torch==1.6.0
torchvision==0.7.0
dgl==0.7.1
scikit-learn==0.23.2
numpy==1.19.1
scipy ==1.5.2
pandas==1.1.2
Dataset
Processed data download link: Dataset
Raw data and more details in JD-Dataset and Alibaba-dataset
Usage
For processed data:
python tb_tmain.py --data_dir ../taobao_data/ --model RHGN --label gender --graph G_ori --gpu 3 # Alibaba-Dataset
python jd_tmain.py --data_dir ../data/ --model RHGN --graph G_ori --label age --gpu 2 # JD-Dataset
How to process data
python process.py --data_dir ../taobao_data/ #Alibaba-Dataset
python process.py --data_dir ../data/ #JD-Dataset
More command can refer run.sh and baseline.sh
