Neural Diffusion Model

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

Training

python train.py

python evaluate.py -model modelfile

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

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

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
data		data
transformer		transformer
DataLoader.py		DataLoader.py
Irvine.chkpt		Irvine.chkpt
Lastfm.chkpt		Lastfm.chkpt
Meme.chkpt		Meme.chkpt
README.md		README.md
evaluate.py		evaluate.py
train.py		train.py