Closed bblodfon closed 6 months ago
Can't reproduce, what versions are you using
library(mlr3verse)
#> Loading required package: mlr3
library(mlr3extralearners)
library(mlr3proba)
set.seed(42)
task = tsk("rats")
part = partition(task, ratio = 0.8)
learner = lrn("surv.ctree")
learner$train(task, part$train)
p = learner$predict(task, part$test)
p$distr[1]$survival(100)
#> [1] 0.733186
p$distr[2]$survival(100)
#> [1] 0.9810414
p$distr[1:2]$survival(100)
#> [,1] [,2]
#> 100 0.733186 0.9810414
Created on 2024-02-23 with reprex v2.0.2
can't either, good!
Created on 2023-11-29 with reprex v2.0.2