Use non-jittered NUTS initialization

This reduces the chance of the "Mass matrix contains zeros on the diagnoal" problem.
JuBiotech · May 20, 2022 · e9a39ff · e9a39ff
1 parent b154234
commit e9a39ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bletl/__init__.py b/bletl/__init__.py
@@ -19,4 +19,4 @@
     NoMeasurementData,
 )
 
-__version__ = "1.1.1"
+__version__ = "1.1.2"
diff --git a/bletl/growth.py b/bletl/growth.py
@@ -146,7 +146,7 @@ def sample(self, **kwargs) -> None:
         sample_kwargs = dict(
             return_inferencedata=True,
             target_accept=0.95,
-            init="jitter+adapt_diag",
+            init="adapt_diag",
             start=self.theta_map,
             tune=500,
             draws=500,