Skip to content

Commit

Permalink
Code to reproduce the experiments in the MLHC paper
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 386124585
  • Loading branch information
abhishekk-ml authored and copybara-github committed Jul 21, 2021
1 parent 78cf564 commit 5f73ffb
Show file tree
Hide file tree
Showing 4 changed files with 35,512 additions and 2 deletions.
9 changes: 9 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -106,6 +106,14 @@ cd agent_based_epidemic_sim/configuration_generator
python3 -m unittest
```

## Risk Score Learning
Jupyter Notebooks for learning risk score models on data sampled from a
discrete grid on \[Infectiousness x Distance x Duration\] are available under
`agent_based_epidemic_sim/learning`. Notebook to reproduce the experiments
in [Murphy et al, Risk score learning for COVID-19 contact tracing apps,
2021][5] is available at
`agent_based_epidemic_sim/learning/MLHC_paper_experiments.ipynb`.

## Acknowledgements

The computational model used by the simulator is inspired in part by the
Expand All @@ -126,3 +134,4 @@ and can be reached at [[email protected]][4].
[2]: https://github.com/BDI-pathogens/OpenABM-Covid19
[3]: https://github.com/GoogleCloudPlatform/covid-19-open-data
[4]: mailto:[email protected]
[5]: https://arxiv.org/abs/2104.08415
4 changes: 2 additions & 2 deletions WORKSPACE
Original file line number Diff line number Diff line change
Expand Up @@ -137,8 +137,8 @@ http_archive(

http_archive(
name = "rules_cc",
strip_prefix = "rules_cc-master",
urls = ["https://github.com/bazelbuild/rules_cc/archive/master.zip"],
strip_prefix = "rules_cc-main",
urls = ["https://github.com/bazelbuild/rules_cc/archive/main.zip"],
)

# GoogleTest/GoogleMock framework. Used by most unit-tests.
Expand Down
Loading

0 comments on commit 5f73ffb

Please sign in to comment.