added handling for factor variables where not all levels are resampled

statimagcoll · Jan 8, 2024 · 4bfd4ed · 4bfd4ed
1 parent f5a857e
commit 4bfd4ed
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/R/internal_functions.R b/R/internal_functions.R
@@ -20,7 +20,8 @@ chisq2Ssq = function(chisq, df, n){
 #' @return Returns a vector of RESI estimates for a single bootstrap replicate
 #' @noRd
 resi_stat = function(dat, inds, mod.full, mod.reduced, boot.method = "nonparam",
-                     cluster = FALSE, clvar = NULL, mod.dat = NULL, overall = TRUE, ...){
+                     cluster = FALSE, clvar = NULL, mod.dat = NULL, overall = TRUE,
+                     nest = NULL, ...){
   # nonparametric bootstrap
   if (boot.method == "nonparam"){
     if(!cluster){
@@ -62,6 +63,11 @@ resi_stat = function(dat, inds, mod.full, mod.reduced, boot.method = "nonparam",
     out = try(resi_pe(mod.full, model.reduced = mod.reduced, data = boot.data, overall = overall, ...)$estimates, silent = T)
     if(inherits(out, "try-error")){
       out = NA
+    } else{
+      # for factored variables, it's possible the model will not contain all factor levels and not represent the original model
+      if(length(out) != nest){
+        out = NA
+      }
     }
   } else{
     out = NA

diff --git a/R/resi.R b/R/resi.R
@@ -209,7 +209,7 @@ resi.default = function(model.full, model.reduced = NULL, data, anova = TRUE,
                   ncpus = ncpus, mod.full = model.full, mod.reduced = model.reduced,
                   anova = anova, coefficients = coefficients, overall = overall, vcovfunc = vcovfunc,
                   Anova.args = Anova.args, vcov.args = vcov.args, unbiased = unbiased,
-                  boot.method = boot.method, clvar = clvar, ...), file = nullfile()))
+                  boot.method = boot.method, clvar = clvar, nest = length(output$estimates), ...), file = nullfile()))
 
   # bootstrapped estimates
   boot.results = boot_out$t