survival_prob_partykit: A wrapper for survival probabilities with partykit models

In censored: '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 model object from partykit::ctree() or partykit::cforest().
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

library(partykit)
c_tree <- ctree(Surv(time, status) ~ age + ph.ecog, data = lung)
survival_prob_partykit(c_tree, lung[1:3, ], eval_time = 100)
c_forest <- cforest(Surv(time, status) ~ age + ph.ecog, data = lung, ntree = 10)
survival_prob_partykit(c_forest, lung[1:3, ], eval_time = 100)

censored documentation built on April 14, 2023, 12:30 a.m.