fix: τ priors for hierarchical Bernoulli models (#86)

TuringLang · Apr 8, 2023 · c52dea3 · c52dea3 · storopoli · Apr 8, 2023
1 parent af9e0ff
commit c52dea3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "TuringGLM"
 uuid = "0004c1f4-53c5-4d43-a221-a1dac6cf6b74"
 authors = ["Jose Storopoli <[email protected]>, Rik Huijzer <[email protected]>, and contributors"]
-version = "2.5"
+version = "2.6"
 
 [deps]
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"

diff --git a/src/turing_model.jl b/src/turing_model.jl
@@ -263,14 +263,14 @@ function _model(μ_X, σ_X, prior, intercept_ranef, idx, ::Type{Bernoulli})
         μ_X=μ_X,
         σ_X=σ_X,
         prior=prior,
-        mad_y=mad(y; normalize=true),
+        std_y=std(y; normalize=true),
     )
         α ~ prior.intercept
         β ~ filldist(prior.predictors, predictors)
         if isempty(intercept_ranef)
             μ = α .+ X * β
         else
-            τ ~ mad_y * truncated(TDist(3); lower=0)
+            τ ~ std_y * truncated(TDist(3); lower=0)
             zⱼ ~ filldist(Normal(), n_gr)
             μ = α .+ τ .* getindex.((zⱼ,), idxs) .+ X * β
         end