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
Note: The code require your gpu memory not less than 20 GB.
Dataset
Raw data and more details in JD-Dataset and Alibaba-dataset
Usage
1、Download processed data:
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
2、Process raw data
python tbdata_process.py #Alibaba-Dataset
python jddata_process.py #JD-Dataset
More command can refer run.sh and baseline.sh