survival_prob_partykit: A wrapper for survival probabilities with partykit models

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

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 model_fit object resulting from decision_tree() with engine = "partykit" or rand_forest() with engine = "partykit".
new_data	A data frame to be predicted.
eval_time	A vector of times to predict the survival probability.
time	Deprecated in favor of eval_time. A vector of times to predict the survival probability.
output	Type of output. Can be either "surv" or "haz".

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)

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