# NeuralDiffusionModel

**Repository Path**: hummingb1rd/NeuralDiffusionModel

## Basic Information

- **Project Name**: NeuralDiffusionModel
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2020-12-23
- **Last Updated**: 2020-12-29

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 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