snowdrop.src.epidemic package
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.