From 4597caa3ae7e45b3202f9220e95e5084c5932d4f Mon Sep 17 00:00:00 2001 From: qiushi Date: Tue, 9 Aug 2022 16:13:39 -0500 Subject: [PATCH] reverse predict class --- R/bart.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R/bart.R b/R/bart.R index 78ee74593..44d80833d 100644 --- a/R/bart.R +++ b/R/bart.R @@ -196,7 +196,7 @@ dbart_predict_calc <- function(obj, new_data, type, level = 0.95, std_err = FALS res <- tibble::tibble(.pred = apply(post_dist, 2, mean, na.rm = TRUE)) } else if (type == "class") { mn <- apply(post_dist, 2, mean, na.rm = TRUE) - lvl <- ifelse(mn > 0.5, obj$lvl[1], obj$lvl[2]) + lvl <- ifelse(mn > 0.5, obj$lvl[2], obj$lvl[1]) lvl <- factor(lvl, levels = obj$lvl) res <- tibble::tibble(.pred_class = lvl) } else if (type == "prob") {