BasisResearch / cities

Home of Basis development for the 2023 TOP Sprint
MIT License
6 stars 0 forks source link

vectorize grouped predictions #105

Closed rfl-urbaniak closed 8 months ago

rfl-urbaniak commented 8 months ago

This adds a new method get_group_predictions to CausalInsightSlim and introduces changes necessary to speed grouped predictions up. Adds test for the functionality and a notebook grouped_predictions.ipynb that illustrates it. At least for now, CausalInsightSlim also preserves the get_fips_predictions functionality, which is useful for backwards compatibility and testing of the new functionality (see test_grouped_predictions.py). Once this has been in use for quite some time without errors, we will consider simplifying and refactoring.