change crit for other gam function

jaredhuling · Sep 3, 2020 · 898e772 · 898e772
1 parent f9e75c8
commit 898e772
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/fit_losses.R b/R/fit_losses.R
@@ -671,8 +671,8 @@ fit_sq_loss_gam <- function(x, y, trt, n.trts, wts, family, match.id, ...)
         var.levels[v] <- length(unique(x[,sel.idx[v]]))
     }
 
-    contin.vars <- sel.vnames[var.levels > 3]
-    binary.vars <- sel.vnames[var.levels <= 3]
+    contin.vars <- sel.vnames[var.levels > 2]
+    binary.vars <- sel.vnames[var.levels <= 2]
 
     # create formula for gam
     contin.formula <- binary.formula <- NULL