survival_prob_pecRpart: A wrapper for survival probabilities with pecRpart models

In EmilHvitfeldt/survnip: 'parsnip' Engines for Survival Models

A wrapper for survival probabilities with pecRpart models

Description

A wrapper for survival probabilities with pecRpart models

Usage

survival_prob_pecRpart(object, new_data, eval_time)

Arguments

object	A parsnip model_fit object resulting from decision_tree() with engine = "rpart".
new_data	Data for prediction.
eval_time	A vector of integers for prediction times.

Value

A tibble with a list column of nested tibbles.

Examples

mod <- decision_tree() %>% 
  set_mode("censored regression") %>%
    set_engine("rpart") %>%
    fit(Surv(time, status) ~ ., data = lung)
survival_prob_pecRpart(mod, new_data = lung[1:3, ], eval_time = 300)

EmilHvitfeldt/survnip documentation built on April 23, 2024, 3:07 p.m.