typo and clear up negbin

mcreel · Apr 23, 2024 · 8aaf761 · 8aaf761
1 parent 9bc59df
commit 8aaf761
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ML/Likelihoods/negbin.jl b/src/ML/Likelihoods/negbin.jl
@@ -9,8 +9,8 @@ function  negbin(θ, y, x, nbtype)
     eps = 1e-8 
     λ = eps .+ exp.(x*β)
     α = eps .+ exp(θ[end])
-    nbtype == 1 ? r = λ./α : r = ones(n)/α
-    p = r ./ (r + λ)
+    nbtype == 1 ? ψ  = λ./α : ψ  = ones(n)/α
+    p = ψ  ./ (ψ  + λ)
     all(r .> 0.0) & all(p .> 0.0) & all(p .< 1.0) ? log.(pdf.(NegativeBinomial.(r, p),y)) : -Inf    
 end