From 02ec0ea2c8c7c77a170144e60c8cd526b2334cc4 Mon Sep 17 00:00:00 2001 From: areichmuth Date: Fri, 14 Apr 2023 14:13:22 +0200 Subject: [PATCH 1/2] Change to continuous input --- R/gbm.step.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R/gbm.step.R b/R/gbm.step.R index 18dfc1b..a11c42b 100644 --- a/R/gbm.step.R +++ b/R/gbm.step.R @@ -118,7 +118,7 @@ gbm.step <- function ( if (is.null(fold.vector)) { if (prev.stratify & family == "bernoulli") { - presence.mask <- data[,gbm.y] == 1 + presence.mask <- data[,gbm.y] > 0 absence.mask <- data[,gbm.y] == 0 n.pres <- sum(presence.mask) n.abs <- sum(absence.mask) From 461663b79af9445a6ed26d2df4f9ae675c39f849 Mon Sep 17 00:00:00 2001 From: areichmuth Date: Fri, 14 Apr 2023 14:14:32 +0200 Subject: [PATCH 2/2] Change to continuous input --- R/gbm.utils.R | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/R/gbm.utils.R b/R/gbm.utils.R index a17cca5..ede396a 100644 --- a/R/gbm.utils.R +++ b/R/gbm.utils.R @@ -22,8 +22,8 @@ stop("obs and preds must be equal lengths") } n.x <- length(obsdat[obsdat == 0]) - n.y <- length(obsdat[obsdat == 1]) - xy <- c(preddat[obsdat == 0], preddat[obsdat == 1]) + n.y <- length(obsdat[obsdat > 0]) + xy <- c(preddat[obsdat == 0], preddat[obsdat > 0]) rnk <- rank(xy) wilc <- ((n.x * n.y) + ((n.x * (n.x + 1))/2) - sum(rnk[1:n.x]))/(n.x * n.y) return(round(wilc, 4))