Skip to content

dsih-artpark/patchsim

Repository files navigation

PatchSim

⚠️ WARNING: Active Development
PatchSim is currently under heavy development. Features are experimental and subject to change. Not recommended for production use at this stage.

PatchSim Banner

PatchSim is a modular metapopulation simulation framework for multi-disease epidemiological modelling.


Vision

To develop a general-purpose, modular simulation framework for patch-based metapopulation epidemiology, enabling modellers and researchers to simulate disease transmission under diverse scenarios, diseases, and intervention strategies. The framework balances robust scientific modelling with flexibility for exploratory research and translational use cases.


Unique Selling Point (USP)

PatchSim combines metapopulation network dynamics with a lightweight, configuration-first workflow:

  • Network-aware compartment transitions across connected patches
  • Arrow-map transition syntax in YAML for explicit model specification
  • Fast iteration loop from config edits to reproducible outputs

Unique Value Proposition (UVP)

Compared to many epidemiology tools that are either code-heavy or tightly bound to a specific disease model, PatchSim offers:

  • Model flexibility: define SIR/SIRS-style variants through config transitions
  • Research-friendly reproducibility: deterministic inputs, logged runs, and versioned config files
  • Dual interface: SDK import for programmatic workflows and CLI for operational runs
  • Extensibility path: built-in structure for adding custom models and project templates

Core Features

PatchSim aims to support a range of modelling features commonly used in metapopulation disease simulations:

  • 🗺️ Spatial Networks: Represent geographical units (e.g., subdistricts, regions) as interconnected patches with movement/contact matrices.
  • 👥 Stratification by Population Attributes:
    • Age groups
    • Species (e.g., cattle, buffalo)
    • Risk groups or occupations
  • 🧪 Disease Agnostic Compartment Models:
    • SIR, SEIR, SIRS and extensions
    • Supports both discrete timestep and ODE-based solvers
  • 🛠️ Scenario and Parameter Management:
    • Batch simulations for scenario comparison
    • Sensitivity analysis and parameter sweeps
  • 🧵 Reproducibility:
    • Random seed control and metadata logging
    • Version-tracked configurations
  • 📦 Modularity:
    • Plug-in architecture for solvers, interventions, and input data pipelines

Installation

Install from PyPI:

pip install patchsim

Install from source using uv:

# Clone the repository
git clone https://github.com/dsih-artpark/patchsim
cd patchsim

# Create a virtual environment and install dependencies
uv venv
source .venv/bin/activate
uv pip install -e .

# For development (with dev dependencies)
uv pip install -e .[dev]

Usage

Command Line Interface

PatchSim provides a subcommand-based CLI. Always run using uv run to ensure correct dependency resolution:

# Show help and available options
uv run patchsim --help

# Show package version
uv run patchsim --version

# Initialize a new self-contained project
uv run patchsim init my-project

# Validate config without running
uv run patchsim validate -c my-project/config.yaml

# Run simulation
uv run patchsim run -c my-project/config.yaml

# List built-in models
uv run patchsim list-models

Python SDK

import patchsim

config = patchsim.load_config("config.yaml")
net, y0, patches, n = patchsim.setup_simulation(config)
patchsim.run_simulation(config, "my-model", net, y0, patches, n)

The simulation outputs are saved in the following structure:

output/
├── logs/
│   └── cli_YYYYMMDD_HHMMSS.log  # Timestamped log files
└── sample-sir-ode/              # Model-specific output directory
    ├── plots/
    │   └── patch_timeseries_sample-sir-ode_ode.png
    └── runs/
        └── all_patches_sample-sir-ode_ode.csv

Configuration

Simulations are configured using YAML files. The configuration file specifies:

  • Model parameters (e.g., transmission rates)
  • Input files (patch populations, seed data, network)
  • Simulation settings (time horizon, output directory)

Example configuration:

# Model parameters
Beta: 0.3
Gamma: 0.1

# Input files
PatchFile: data/patch/sample-sir-ode-patch-population.csv
SeedFile: data/seeds/sample-sir-ode-patchA-2.csv
NetworkFile: data/networks/sample-network-static.csv

# Simulation settings
OutputDir: output/sample-sir-ode
TMax: 50

Contributing

We welcome contributions!

To contribute: fork the repo, create a branch, make your changes, and open a pull request.
For major changes, please open an issue first to discuss.

Thanks for helping improve the framework!

License

This project is licensed under the GNU General Public License v3.0.

License: GPL v3

You may use, modify, and share this project under the same license terms. See the LICENSE file for full details.

About

PatchSim is a modular metapopulation simulation framework for multi-disease epidemiological modelling.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages