zfjsail / wechat-wow-analysis

Code for our TKDE paper "Understanding WeChat User Preferences and “Wow” Diffusion"
MIT License
19 stars 1 forks source link

wechat-wow-analysis

Understanding WeChat User Preferences and “Wow” Diffusion.
Fanjin Zhang, Jie Tang, Xueyi Liu, Zhenyu Hou, Yuxiao Dong, Jing Zhang, Xiao Liu, Ruobing Xie, Kai Zhuang, Xu Zhang, Leyu Lin, and Philip S. Yu.
TKDE 2021 (accepted)

Prerequisites

Getting Started

Installation

Clone this repo.

git clone https://github.com/zfjsail/wechat-wow-analysis.git
cd wechat-wow-analysis

Please install dependencies by

pip install -r requirements.txt

Dataset

Due to data privacy issue in WeChat, we provide a public Weibo dataset to evaluate our prediction model. The Weibo dataset can be downloaded from Baidu Pan. Unzip the file and put the weibo directory into DATA_DIR specified in settings.py.

How to run

cd $project_path
export CUDA_VISIBLE_DEVICES='?'  # specify which GPU(s) to be used
python train.py

Cite

Please cite our paper if you use this code in your own work:

@article{zhang2021understanding,
  title={Understanding WeChat User Preferences and “Wow” Diffusion.},
  author={Zhang, Fanjin and Tang, Jie and Liu, Xueyi and Hou, Zhenyu and Dong, Yuxiao and Zhang, Jing and Liu, Xiao and Xie, Ruobing and Zhuang, Kai and Zhang, Xu and Lin, Leyu and Yu, Philip.},
  journal={IEEE Transactions on Knowledge and Data Engineering},
  year={2021}
}