drop = false for 1 predictor

ropensci · May 2, 2024 · ded908b · ded908b
1 parent c1fba42
commit ded908b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/orsf_R6.R b/R/orsf_R6.R
@@ -3051,15 +3051,14 @@ ObliqueForest <- R6::R6Class(
      unique(variable[predictor %in% colnames(cpp_args$x)])
     )
 
-
     oob_data[n_predictors,
              `:=`(n_predictors = n_predictors,
                   stat_value = cpp_output$eval_oobag$stat_values[1,1],
                   variables_included = .variables_included,
                   predictors_included = colnames(cpp_args$x),
                   predictor_dropped = worst_predictor)]
 
-    cpp_args$x <- cpp_args$x[, -worst_index]
+    cpp_args$x <- cpp_args$x[, -worst_index, drop = FALSE]
     n_predictors <- n_predictors - 1
     current_progress <- current_progress + 1
 
@@ -3069,6 +3068,7 @@ ObliqueForest <- R6::R6Class(
     cat("Selecting variables: 100%\n")
    }
 
+
    collapse::na_omit(oob_data)
 
   },