X-Git-Url: https://git.auder.net/?a=blobdiff_plain;f=R%2FR6_AgghooCV.R;h=2734d69b48d1ce30811a940f09d67ef59c3d4a72;hb=1fdc3c344600de5932439a434abeff4b14e6591f;hp=9cdf19ed5081aec44fc3e4ceb9a72dcb7db306b3;hpb=504afaadc783916dc126fb87ab9e067f302eb2c5;p=agghoo.git diff --git a/R/R6_AgghooCV.R b/R/R6_AgghooCV.R index 9cdf19e..2734d69 100644 --- a/R/R6_AgghooCV.R +++ b/R/R6_AgghooCV.R @@ -98,7 +98,7 @@ AgghooCV <- R6::R6Class("AgghooCV", all_predictions[,v] <- private$pmodels[[v]]$model(X) if (private$task == "regression") # Easy case: just average each row - rowSums(all_predictions) + return (rowMeans(all_predictions)) # "Hard" classification: apply(all_predictions, 1, function(row) { t <- table(row)