Correct error in notebook with fixed C, all data is now used for like…

…lihood
Priesemann-Group · May 26, 2021 · b970c7b · b970c7b
1 parent 116260a
commit b970c7b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/covid19_npis/model/__init__.py b/covid19_npis/model/__init__.py
@@ -18,7 +18,7 @@
     convolution_with_map,
 )
 
-from .approximate_posterior import build_approximate_posterior
+from .approximate_posterior import build_approximate_posterior, build_iaf
 
 from . import number_of_tests
 from . import deaths

diff --git a/scripts/notebooks/SIR_model_with_age_groups_fixed_C.ipynb b/scripts/notebooks/SIR_model_with_age_groups_fixed_C.ipynb
@@ -239,6 +239,7 @@
     "modelParams._dataframe_new_cases = positive_tests[1:] \n",
     "modelParams.pos_tests_data_tensor = positive_tests[1:]\n",
     "positive_tests_formatted = modelParams.pos_tests_data_tensor[1:]\n",
+    "modelParams._set_data_mask()\n",
     "this_model = SIR_model(modelParams) \n",
     "\n",
     "print(modelParams.pos_tests_data_tensor[1:6])\n",