Add optional plotting of trajectories from posterior

seananderson / covidseir

Bayesian SEIR model to estimate the effects of social-distancing on COVID-19

https://seananderson.github.io/covidseir/index.html

28 stars 11 forks source link

Add optional plotting of trajectories from posterior #8

Open sempwn opened 3 years ago

sempwn commented 3 years ago

Add option to plot_projection to output a sample of trajectories from the posterior as opposed to showing the ribbons representing the percentiles.

seananderson commented 3 years ago

I was thinking it could be a sample of 3-20 posterior lines on top of the ribbon. Maybe make an argument that defaults to 0 that represents the number to overlay?

An example:

It's probably too much for public consumption, but might be useful for internal use.

sempwn commented 3 years ago

I like that approach. Might want additional options to play around with the alpha of the ribbons and the sampled trajectories as could end up looking a little crowded in some figures.