Code Monkey home page Code Monkey logo

mechbayes's Introduction

Bayesian models for COVID-19

This repository contains code for Bayesian modeling of COVID-19 using numpyro and jax.

Models

Team

The team is based at the College of Information and Computer Sciences and School of Public Health & Health Sciences at UMass. The model creators, contributors, and weekly crank-turners are:

Dr. Reich and Dr. Ray direct the CDC Influenza Forecasting Center of Excellence at UMass.

Installation

Our code depends on numpyro and jax. If you don't have these packages, our installation routine will pull and install them:

git clone https://github.com/dsheldon/mechbayes
cd mechbayes
pip install -e .

Installation Details

If you need to manually install jax and numpyro, please see the instructions for those packages: jax, numpyro.

mechbayes's People

Contributors

dsheldon avatar elray1 avatar nickreich avatar nutchaw avatar serena-wang avatar

Stargazers

 avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

mechbayes's Issues

add support for getting hospitalizations data

to start, only support US states and national, getting data from the covidcast API.

For states:

  • rename jhu to something more general
  • add argument to jhu.load_us to specify which data source to load from (jhu or covidcast)
  • add a utility function similar to load_us_time_series that gets cases, hospitalizations, or deaths data from covidcast for all states. Function arguments are target_variable ("cases", "hospitalizations", or "deaths"), and an optional as_of parameter. It should return a pandas data frame with the date as an index and one column for each state-level location. The column names should match the location names returned by jhu.get_state_info(). The values in the columns will be the case/hospitalization/death counts (just one for a single call to the function) in each state on the given date.
  • At

    mechbayes/mechbayes/jhu.py

    Lines 145 to 146 in c51d090

    confirmed = load_us_time_series("time_series_covid19_confirmed_US.csv")
    deaths = load_us_time_series("time_series_covid19_deaths_US.csv")
    , condition on the data source to either do the existing calls to load_us_time_series or call the new function above. For now, when calling the utility function for covidcast, just omit the as_of parameter to get the latest available data, matching current behavior for cases and deaths. If data source is covidcast, add hosps to the concat call a couple lines down.

For US national:

  • add an entry to the dictionary at
    d = {key: load_and_massage(url) for key, url in sources.items()}
    for hosp. For now, it should have observed value for the US only, and nan's for all other locations

For counties:

  • we don't have observations for counties, add a column with all nan's

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.