fit_survival_forest: Wrapper of 'grf::survival_forest'

View source: R/fit_surv.R

fit_survival_forestR Documentation

Wrapper of grf::survival_forest

Description

Wrapper of grf::survival_forest

Usage

fit_survival_forest(
  formula,
  data,
  id.var,
  time.var,
  event.var,
  nfold = 1,
  oob = TRUE,
  option = list(),
  ...
)

Arguments

formula

formula used by grf::survival_forest

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

oob

whether to use out-of-bag (OOB) fitted values from grf::survival_forest when sample splitting is not used (nfold=1). Default is TRUE

option

a list containing optional arguments passed to grf::survival_forest. We encourage using a named list. Will be passed to grf::survival_forest by running a command like do.call(survival_forest, option). The user should not specify X, Y, D, failure.times and compute.oob.predictions.

...

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.