fit_rpart: Wrapper of 'rpart::rpart' and 'partykit::as.party'

In QIU-Hongxiang-David/SDRsurv: Sequentially Doubly Robust Estimation of Survival Curve with Time-Varying Covariates

Wrapper of rpart::rpart and partykit::as.party

Description

First use rpart::rpart to obtain a tree and then use as.party to obtain Kaplan-Meier fits.

Usage

fit_rpart(
  formula,
  data,
  id.var,
  time.var,
  event.var,
  nfold = 1,
  option = list(),
  ...
)

Arguments

formula	formula used by rpart::rpart
data	data containing all covariates, follow-up time, event indicator and id
id.var	see SDRsurv
time.var	see SDRsurv
event.var	see SDRsurv
nfold	number of folds used when fitting survival curves with sample splitting. Default is 1, meaning no sample splitting
option	a list containing optional arguments passed to rpart::rpart. We encourage using a named list. Will be passed to rpart::rpart by running a command like do.call(rpart, option). The user should not specify formula and data.
...	ignored

Value

a pred_surv class containing fitted survival curves for individuals in data

QIU-Hongxiang-David/SDRsurv documentation built on March 29, 2024, 8:41 a.m.