Source code for CIKM 2021 paper for Relation-aware Heterogeneous Graph for User Profiling
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
Raw data and more details in JD-Dataset and Alibaba-dataset
Processed data download link: Dataset
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
python tbdata_process.py #Alibaba-Dataset
python jddata_process.py #JD-Dataset
More command can refer run.sh and baseline.sh