add_surv_prob2: Add survival probability estimates

In adibender/pem.xgb: GBT (PEM)

Description

Given suitable data (i.e. data with all columns used for estimation of the model), this functions adds a column surv_prob containing survival probabilities for the specified covariate and follow-up information.

Usage

add_surv_prob2(
  newdata,
  object,
  overwrite = FALSE,
  time_var = NULL,
  interval_length = "intlen",
  ...
)

Arguments

newdata	A data frame or list containing the values of the model covariates at which predictions are required. If this is not provided then predictions corresponding to the original data are returned. If newdata is provided then it should contain all the variables needed for prediction: a warning is generated if not. See details for use with link{linear.functional.terms}.
object	a fitted gam object as produced by gam().
overwrite	Should hazard columns be overwritten if already present in the data set? Defaults to FALSE.
time_var	Name of the variable used for the baseline hazard. If not given, defaults to "tend" for gam fits, else "interval". The latter is assumed to be a factor, the former numeric.
interval_length	The variable in newdata containing the interval lengths. Can be either bare unquoted variable name or character. Defaults to "intlen".
...	Further arguments passed to predict.xgb.Booster.

adibender/pem.xgb documentation built on Sept. 10, 2021, 7:24 p.m.