add more doc for 'which.curve'

mlr-org · Oct 16, 2023 · 1e66dc9 · 1e66dc9
1 parent 4204c2d
commit 1e66dc9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/R/learner_BART_surv_bart.R b/R/learner_BART_surv_bart.R
@@ -22,7 +22,7 @@
 #' @section Custom mlr3 parameters:
 #' - `quiet` allows to suppress messages generated by the wrapped C++ code. Is initialized to `TRUE`.
 #' - `importance` allows to choose the type of importance. Default is `count`, see documentation of method `$importance()` for more details.
-#' - `which.curve` allows to choose which posterior draw will be used for the calculation of the `crank` prediction. By default the **median posterior** is used.
+#' - `which.curve` allows to choose which posterior draw will be used for the calculation of the `crank` prediction. If between (0,1) it is taken as the quantile of the curves otherwise if greater than 1 it is taken as the curve index, can also be 'mean'. By default the **median posterior** is used, i.e. `which.curve` is 0.5.
 #'
 #' @templateVar id surv.bart
 #' @template learner

diff --git a/man/mlr_learners_surv.bart.Rd b/man/mlr_learners_surv.bart.Rd