From 9347b7ba0f684b0bbe2a3fefcd171a408f96a733 Mon Sep 17 00:00:00 2001 From: perrydv Date: Fri, 12 Jun 2026 09:00:24 -0700 Subject: [PATCH] update numDivergences in sampler_NUTS --- nimbleHMC/R/HMC_samplers.R | 1 + 1 file changed, 1 insertion(+) diff --git a/nimbleHMC/R/HMC_samplers.R b/nimbleHMC/R/HMC_samplers.R index 9da05c5..9836e3a 100644 --- a/nimbleHMC/R/HMC_samplers.R +++ b/nimbleHMC/R/HMC_samplers.R @@ -1078,6 +1078,7 @@ sampler_NUTS <- nimbleFunction( if(is.nan(new_H)) new_H <- Inf deltaH <- new_H - H0 if(deltaH > deltaMax) divergent <<- TRUE + if(divergent) numDivergences <<- numDivergences + 1 branch$log_sum_wt <- log_sum_exp(branch$log_sum_wt, -deltaH) if((-deltaH) > 0) sum_metropolis_prob <<- sum_metropolis_prob + 1 else sum_metropolis_prob <<- sum_metropolis_prob + exp(-deltaH)