fix threshold in print

stan-dev · Jan 26, 2024 · 6565f45 · 6565f45
1 parent 61c0725
commit 6565f45
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/R/print.R b/R/print.R
@@ -39,7 +39,9 @@ print.psis_loo <- function(x, digits = 1, plot_k = FALSE, ...) {
   print.loo(x, digits = digits, ...)
   cat("------\n")
   print_mcse_summary(x, digits = digits)
-  if (length(pareto_k_ids(x, threshold = 0.7))) {
+  S <- dim(x)[1]
+  k_threshold <- ps_khat_threshold(S)
+  if (length(pareto_k_ids(x, threshold = k_threshold))) {
     cat("\n")
   }
   print(pareto_k_table(x), digits = digits)
@@ -65,7 +67,9 @@ print.psis_loo_ap <- function(x, digits = 1, plot_k = FALSE, ...) {
   cat("------\n")
   cat("Posterior approximation correction used.\n")
   print_mcse_summary(x, digits = digits)
-  if (length(pareto_k_ids(x, threshold = 0.7))) {
+  S <- dim(x)[1]
+  k_threshold <- ps_khat_threshold(S)
+  if (length(pareto_k_ids(x, threshold = k_threshold))) {
     cat("\n")
   }
   print(pareto_k_table(x), digits = digits)