Skip to content

Latest commit

 

History

History
executable file
·
40 lines (35 loc) · 882 Bytes

README.md

File metadata and controls

executable file
·
40 lines (35 loc) · 882 Bytes

Accelerating Stochastic Simulation with Interactive Neural Processes

This repository is a PyTorch implementation of Accelerating Stochastic Simulation with Interactive Neural Processes.

Requirements

  • torch
  • scipy>=0.19.0
  • numpy>=1.12.1
  • pandas>=0.19.2
  • pyyaml
  • statsmodels
  • tensorflow>=1.3.0
  • torch
  • tables
  • future
  • sklearn
  • matplotlib
  • gpytorch
  • math

To install requirements:

pip install -r requirements.txt

Model Training and Evaluation for SEIR Simulator

cd 2d_seir_code/*
python main.py

Dataset for LEAM-US Simulator

Download test_data to dataset_dir: data/data/test_data

Model Training and Evaluation for LEAM-US Simulator

cd leam_us_code/STNP/*
python dcrnn_train_pytorch.py
cd leam_us_code/offline/*
python dcrnn_train_pytorch.py