estimate overdispersion in hospital_admissions_model.qmd

CDCgov · Jul 15, 2024 · 673bab2 · 673bab2
1 parent 5e6bcc7
commit 673bab2
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/docs/source/tutorials/hospital_admissions_model.qmd b/docs/source/tutorials/hospital_admissions_model.qmd
@@ -176,7 +176,15 @@ rtproc = process.RtRandomWalkProcess(
 )
 
 # The observation model
-obs = observation.NegativeBinomialObservation(concentration_prior=1.0)
+obs = observation.NegativeBinomialObservation(
+    "negbinom_rv",
+    metaclass.DistributionalRV(
+        dist.TransformedDistribution(
+            dist.HalfNormal(), transformation.PowerTransform(-2)
+        ),
+        "concentration",
+    ),
+)
 ```
 
 Notice all the components are `RandomVariable` instances. We can now build the model: