line switch preserves n

bips-hb · Nov 28, 2023 · 74b4661 · 74b4661
1 parent f25f26e
commit 74b4661
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/utils.R b/R/utils.R
@@ -167,8 +167,8 @@ leaf_posterior <- function(params, evidence) {
   # Continuous features
   if (any(evidence$family != 'multinom')) { 
     evi <- evidence[family != 'multinom']
-    psi <- merge(evi, params$cnt, by = 'variable')
     evi[, n := .N, by = variable]
+    psi <- merge(evi, params$cnt, by = 'variable')
     if (any(evi$relation == '==')) {
       psi[relation == '==', lik := 
             truncnorm::dtruncnorm(value, a = min, b = max, mean = mu, sd = sigma)]