snowdrop.src.epidemic package

Submodules

snowdrop.src.epidemic.covid19 module

Created on Tue Jun 12 18:50:05 2018 Compares results of a baseline and an alternative scenarios runs. @author: agoumilevski

snowdrop.src.epidemic.covid19.div(df, d)[source]
snowdrop.src.epidemic.covid19.hp(x)[source]
snowdrop.src.epidemic.covid19.plot(df1, df2, countries, titles, labels, fname)[source]

Plot 1D graphs.

snowdrop.src.epidemic.covid19.readData(file_path)[source]

Read csv file and returns pandas dataframe.

Parameters:
param file_path:

Path to excel file.

type file_path:

str.

returns:

pandas dataframe.

snowdrop.src.epidemic.covid19.run(f1, f2, countries, population)[source]

Compare results.

snowdrop.src.epidemic.data module

Created on Wed Jan 20 23:39:29 2021

@author: alexei

snowdrop.src.epidemic.data.annualize(data)[source]

Annualizes growth rate.

snowdrop.src.epidemic.data.filter_solution(data, lmbd=1600)[source]

Filter data.

snowdrop.src.epidemic.data.getData(start, end, country=None, save=True)[source]

Read excel file.

snowdrop.src.epidemic.data.getData2(start, end, sht, save=True)[source]

Read excel file.

snowdrop.src.epidemic.data.make_space_above(fig, topmargin=1)[source]

increase figure size to make topmargin (in inches) space for titles, without changing the axes sizes

snowdrop.src.epidemic.data.plot_data(db, labels, save=True, size=[3, 3], show=True)[source]

snowdrop.src.epidemic.ert module

Created on Monday December 28, 2020 @author: A.Goumilevski

This code solves and simulates the model developed by Eichenbaum M.S., Rebelo S., Trabandt M. (2020) in “Epidemics in the Neoclassical and New Keynesian Models”. NBER Working Paper 27430, http://www.nber.org/papers/w27430

snowdrop.src.epidemic.ert.SIR(pii, pir, pid, s0, i0, r0, dd0, pop0, lockdown_rate=None, vaccination_rate=None)[source]

SIR model forecast.

snowdrop.src.epidemic.ert.calibrate(pi1, pi2, pi3, pi1_shr_target, pi2_shr_target)[source]

Calibrate pi1, pi2 and pi3 using SIR model.

snowdrop.src.epidemic.ert.func(x)[source]

Calculate infection spread.

snowdrop.src.epidemic.ert.func1(x)[source]

Calibration function.

snowdrop.src.epidemic.ert.func2_squared(x)[source]

Error function for constrained calibration.

snowdrop.src.epidemic.ert.func_squared(x)[source]

Error function for constrained calibration.

snowdrop.src.epidemic.ert.getYearlySeries(series, fpath, sheet_name, column, startrow, save=False)[source]

Convert weekly frequency to yearly frequency.

This approximates shortage of labor supply due to COVID-19 effects. It is used in DIGNAR-19 toolkitf to calibrate model.

snowdrop.src.epidemic.ert.integral(betta, theta, S, I, R, cs, ci, cr, ns, ni, nr, Ti, income)[source]

Compute household lifetime utility.

Cost of loss of lives is a discounted value of ‘Ti’ years of individual income.

snowdrop.src.epidemic.ert.main(Plot=False, save=True)[source]

Forecast based on ERT model.

snowdrop.src.epidemic.gsw module

Created on Thu Jan 28 00:25:58 2021

@author: A.Goumilevski

snowdrop.src.epidemic.gsw_tunes module

snowdrop.src.epidemic.gsw_v2 module

Created on Thu Jan 28 00:25:58 2021

@author: Alexei

snowdrop.src.epidemic.sir_params module

Created on Wed Mar 24 17:01:29 2021

@author: alexei

snowdrop.src.epidemic.sir_params.SIR(x)[source]

SIR model.

snowdrop.src.epidemic.sir_params.SIRS(x)[source]

SIR model with wildtype and resistant virus strain.

snowdrop.src.epidemic.sir_params.SIRV(x)[source]

SIR model with two virus strains and vaccination.

snowdrop.src.epidemic.sir_params.SIRVS(x)[source]

SIR model with wildtype and resistant virus strain.

snowdrop.src.epidemic.sir_params.Virus(x)[source]
snowdrop.src.epidemic.sir_params.calibrate(x0)[source]

Calibrate pi1, pi2 and pi3 using SIR model.

snowdrop.src.epidemic.sir_params.filter_solution(data, lmbd=1600)[source]

Filter data.

snowdrop.src.epidemic.sir_params.func(x)[source]

Calibration function.

snowdrop.src.epidemic.sir_params.func_squared(x)[source]

Error function for constrained calibration.

snowdrop.src.epidemic.sir_params.getData(start, end, country=None, save=True)[source]

Read data files.

snowdrop.src.epidemic.sir_params.plot_data(db, country, fname=None, save=True, show=True, legend=None, sizes=[1, 1])[source]

Plots data.

Module contents