Fix testset indicator bug in boosting classification

jasp-stats · May 24, 2024 · 8097181 · 8097181
1 parent 387791c
commit 8097181
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/mlClassificationBoosting.R b/R/mlClassificationBoosting.R
@@ -93,7 +93,7 @@ mlClassificationBoosting <- function(jaspResults, dataset, options, ...) {
   testIndicatorColumn <- rep(1, nrow(dataset))
   testIndicatorColumn[trainingIndex] <- 0
   # gbm expects the columns in the data to be in the same order as the variables...
-  trainingAndValidationSet <- trainingAndValidationSet[, match(names(trainingAndValidationSet), all.vars(formula)), drop = FALSE]
+  trainingAndValidationSet <- trainingAndValidationSet[, match(names(trainingAndValidationSet)[which(names(trainingAndValidationSet) %in% all.vars(formula))], all.vars(formula)), drop = FALSE]
   if (options[["modelOptimization"]] == "manual") {
     # Just create a train and a test set (no optimization)
     trainingSet <- trainingAndValidationSet