A package for compartmental epidemic modelling
Project description
Пакет epidemmo
Пакет для создания эпидемиологических моделей. Структура создаваемых моделей основана на идеях компартментального моделирования.
Импорт пакета
import epidemmo
Создание простой SIR модели
from epidemmo import ModelBuilder
from matplotlib import pyplot as plt
builder = ModelBuilder()
builder.add_stage('S', 100).add_stage('I', 1).add_stage('R')
builder.add_factor('beta', 0.4).add_factor('gamma', 0.1)
builder.add_flow('S', 'I', 'beta', 'I').add_flow('I', 'R', 'gamma')
model = builder.build()
result_df = model.start(70)
result_df.plot(title='SIR', ylabel='population', xlabel='time')
plt.show()
start(70)
- метод, который принимает длительность моделирования, а возвращает pd.DataFrame с результатами моделирования.
Результаты моделирования
Использование стандартных моделей
Пакет содержит несколько стандартных эпидемиологических моделей.
from epidemmo import Standard
model = Standard.get_SIR_builder().build()
result = model.start(40)
Вы можете изменить стартовую численность каждой стадии, а также изменить значение параметров модели.
from epidemmo import Standard
model = Standard.get_SIR_builder().build()
model.set_start_stages(S=1000, I=10, R=0)
model.set_factors(beta=0.5)
Вывод и запись табличных результатов
После запуска модели Вы можете вывести результаты в виде таблицы (PrettyTable) в консоль.
from epidemmo import Standard
model = Standard.get_SIR_builder().build()
model.start(60)
model.print_result_table()
или записать результаты в csv файлы, включая
- файл с изменением численности каждой стадии
- файл с изменением значений всех параметров во времени
- файл с изменением интенсивности потоков модели во времени
from epidemmo import Standard
model = Standard.get_SIR_builder().build()
model.start(60)
model.write_results()
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
epidemmo-0.0.8.tar.gz
(46.1 kB
view hashes)
Built Distribution
epidemmo-0.0.8-py3-none-any.whl
(13.7 kB
view hashes)