fitBARTSurvival: fitBARTSurvival
In marioni-group/MethylPipeR: MethylPipeR

fitBARTSurvival

R Documentation

fitBARTSurvival

Description

fitBARTSurvival

Usage

fitBARTSurvival(
  trainXs,
  trainTarget,
  testXs,
  pipelineRunInformation,
  sparse = FALSE,
  mcCores = 10,
  nTree = 100,
  k = 2,
  nSkip = 500,
  type = "pbart"
)

Arguments

`trainXs`	A matrix/data.frame of X variables to be used in the model fitting. Columns should correspond to variables.
`trainTarget`	A matrix/data.frame including columns named 'Event' and 'time_to_event' corresponding to a binary indicator of event/censoring and the time to event/censoring respectively.
`testXs`	A matrix/data.frame of out-of-sample data to be used in model evaluation. The set of variables should be the same as those in `trainXs`.
`pipelineRunInformation`	The object originally created by `beginPipelineRun`.
`sparse`	Set to `TRUE` if the sparsity inducing Dirichlet prior should be used for splitting rule variable selection.
`mcCores`	The number of cores to be used by the MCMC sampler.
`nTree`	The number of trees to be used by BART.
`k`	The value of the hyperparameter k.
`nSkip`	The number of MCMC samples that should be treated as burn-in and discarded.
`type`	`'pbart'` (probit) or `'lbart'` (logit)