Merge pull request #29 from Kyoshido/Kyoshido-patch-2

address issue 22 by @Kyoshido
statistikat · Aug 4, 2023 · 701228c · 701228c
2 parents 4b7bd5b + b9a18cb
commit 701228c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/R/simCategorical.R b/R/simCategorical.R
@@ -64,6 +64,7 @@ generateValues <- function(dataSample, dataPop, params) {
       probs <- predict(mod, newdata=newdata, type="probs")
     }else if ( meth %in% c("ctree","cforest") ) {
       probs <- predict(mod, newdata=data.table(newdata), type="prob")
+      probs <- split(probs, seq(nrow(probs)))  
       probs <- do.call("rbind",probs)
 	    if(ncol(probs)==2){
         probs <- probs[,2]