survival_prob_partykit {censored} | R Documentation |
A wrapper for survival probabilities with partykit models
Description
A wrapper for survival probabilities with partykit models
Usage
survival_prob_partykit(
object,
new_data,
eval_time,
time = deprecated(),
output = "surv"
)
Arguments
object |
A parsnip |
new_data |
A data frame to be predicted. |
eval_time |
A vector of times to predict the survival probability. |
time |
Deprecated in favor of |
output |
Type of output. Can be either |
Value
A tibble with a list column of nested tibbles.
Examples
tree <- decision_tree() %>%
set_mode("censored regression") %>%
set_engine("partykit") %>%
fit(Surv(time, status) ~ age + ph.ecog, data = lung)
survival_prob_partykit(tree, lung[1:3, ], eval_time = 100)
forest <- rand_forest() %>%
set_mode("censored regression") %>%
set_engine("partykit") %>%
fit(Surv(time, status) ~ age + ph.ecog, data = lung[1:100, ])
survival_prob_partykit(forest, lung[1:3, ], eval_time = 100)
[Package censored version 0.3.3 Index]