- Main.py - Entry file to the project, running set of simulation to generate the results shown in the paper.
- edge.py - Data Structure class, edge object of the graph object.
- epidemiological_state.py - Enum class for the SEIIRRD epidemiological model.
- graph.py - Data Structure class, a classical graph object.
- multi_sim.py - A technical class to run multiple instances of the same Simulator object.
- node.py - Data Structure class, node object of the graph object and operating as individual in the population.
- params.py - Enum class for the simulator's parameter values.
- plotter.py - A plots generator central class.
- sim.py - The main class in the project, responsible to run the simulation.
- sim_generator.py - A class responsible to generate random instances with some pre-defined properties of the simulator.
- math_utils.py - pre-compiled computational funcions.
- vaccine_reduction.py - the infection factor reduction due to vaccine reduction.
Please cite the the paper assosited to this work if you use it in any academic paper:
@article{lazebnik2022,
title={Early Detection and Control of the Next Epidemic Wave Using Health Communications: Development of an Artificial Intelligence-Based Tool and Its Validation on COVID-19 Data from the US},
author={Lazebnik, T. and Bunimovich-Mendrazitsky, S. and Ashkenazi, S. and Levner, E. and Benis, A.},
journal={Int. J. Environ. Res. Public Health },
year={2022},
volume = {19},
number = {23},
pages = {16023},
doi = {10.3390/ijerph192316023}
}
- Python 3.7
- numpy 1.20.2
- matplotlib 3.4.0
- pandas 1.2.3
- scipy 1.7.0
- numba 0.55.1
- seaborn 0.11.2
- NetworkX 2.6.2
- scikit-learn 1.0.2
- tuna Latest
- line_profiler Latest
- Flask Latest
- Werkzeug Latest
- yagmail Latest
These can be found in the requirements.txt and easily installed using the "pip install requirements.txt" command in your terminal.
- Clone the repo
- Install the 'requirements.txt' file (pip install requirements.txt)
- Run the 'main.py' file (python main.py or python3 main.py)
- Checkout the results in the "results" folder.