stop if multiclass

egenn · Jul 1, 2024 · e0f7f91 · e0f7f91
1 parent 9cf09e3
commit e0f7f91
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/R/s_LightRuleFit.R b/R/s_LightRuleFit.R
@@ -114,6 +114,10 @@ s_LightRuleFit <- function(x, y = NULL,
   xnames <- dt$xnames
   type <- dt$type
   checkType(type, c("Classification", "Regression"), mod.name)
+  # Currently only supports binary classification
+  if (type == "Classification" && length(levels(y)) > 2) {
+    stop("Only binary classification is currently supported")
+  }
   # if (any(sapply(x, is.factor))) {
   #   factor_index <- names(x)[which(sapply(x, is.factor))]
   #   xp <- preprocess(x,
@@ -286,6 +290,8 @@ s_LightRuleFit <- function(x, y = NULL,
     n.cores = n.cores
   )
   mod_glmnet_select <- do.call(s_GLMNET, glmnet_select_args)
+  # binary classification: coef(mod_glmnet_select$mod) is n x 1 sparse Matrix of class "dgCMatrix"
+  # multiclass classification
   rule_coefs <- data.matrix(coef(mod_glmnet_select$mod))
   intercept_coef <- rule_coefs[1, , drop = FALSE]
   colnames(intercept_coef) <- "Coefficient"