Neural Diffusion Model

source code for TKDE submission "Neural Diffusion Model for Microscopic Cascade Prediction"

Training

python train.py

Test

python evaluate.py -model modelfile

Parameters

Parameters can be modified in DataLoader.py and train.py

Requirements

pyTorch==0.2.0 and A GPU device for training and testing

Contact

cheng-ya14@mails.tsinghua.edu.cn