https://github.com/adityalab/h2abm
Code & Demo for SDM24 Submission: H^2ABM: Heterogeneous Agent-based model on Hypergraphs to Capture Group Interactions
https://github.com/adityalab/h2abm
Last synced: 28 days ago
JSON representation
Code & Demo for SDM24 Submission: H^2ABM: Heterogeneous Agent-based model on Hypergraphs to Capture Group Interactions
- Host: GitHub
- URL: https://github.com/adityalab/h2abm
- Owner: AdityaLab
- License: mit
- Created: 2023-09-22T18:29:38.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2024-04-22T19:01:38.000Z (about 1 year ago)
- Last Synced: 2024-04-22T20:27:32.784Z (about 1 year ago)
- Language: Python
- Size: 873 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# [SDM 2024] H2ABM: Heterogeneous Agent-based Model on Hypergraphs to Capture Group Interactions (Best Poster Award)
## Publication
Implementation of the paper "H2ABM: Heterogeneous Agent-based Model on Hypergraphs to Capture Group Interactions."
Authors: Vivek Anand*, Jiaming Cui*, Jack Heavey, Anil Vullikanti, B. Aditya Prakash
*Equal contribution
Venue: SDM 2024
Link to the paper: https://sites.cc.gatech.edu/~jcui75/papers/h2abm-sdm24.pdf
We also release an anonymized version of the UVA dataset at: https://github.com/AdityaLab/UVA-Hypergraph/
## Setup
First install Anaconda. The dependencies are listed in `environment.yml` file.
Then run the following commands:
```bash
conda env create --prefix ./envs/hhabm --file environment.yml
source activate ./envs/hhabm
```
## Directory structure
```
-data
- synthetic.pkl -> save synthetic hypergraphs as pkl file
- synthetic.csv -> save synthetic number of cases as csv file
- model.py -> implementation of Hypergraph-HeterSIS model
- calibrateh.py -> calibrate the Hypergraph-HeterSIS model to the synthetic number of cases
- simulateh.py -> run simulations for the Hypergraph-HeterSIS model based on the calibrated parameters
- evaluate.py -> Calculate the NRMSE, ND, and Pearson correlation for Hypergraph-HeterSIS model results
- outputs -> save results
```
## Dataset
The dataset is at `data` folder. It contains the hypergraph file (synthetic.pkl) and the number of cases (synthetic.csv) used for Hypergraph-HeterSIS model calibration.
## Demo
We provde a demo code to calibrate the Hypergraph-HeterSIS model and calculate the metrics we used in main article
Run:
```
chmod 777 run.sh
./run.sh
```
This will save the results (NRMSE, ND, and Pearson correlation value) in `output/result.csv`
## Cite our work
If you find our work useful, please cite our work:
- Vivek Anand*, Jiaming Cui*, Jack Heavey, Anil Vullikanti, B. Aditya Prakash. H2ABM: Heterogeneous Agent-based Model on Hypergraphs to Capture Group Interactions. SDM 2024.
```
@inproceedings{anand2024h2abm,
title={},
author={Anand, Vivek and Cui, Jiaming and Heavey, Jack and Vullikanti, Anil and Prakash, B Aditya},
booktitle={Proceedings of the 2024 SIAM International Conference on Data Mining (SDM)},
pages={280--288},
year={2024},
organization={SIAM}
}
```