Skip to content

albertyang33/NeuralDiffusionModel

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

[email protected]

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages