R: Подгонка деревьев выживания с изменяющимися во времени ковариатами в RandomForestSRC - PullRequest
Новая
       25

R: Подгонка деревьев выживания с изменяющимися во времени ковариатами в RandomForestSRC

0 голосов
/ 07 марта 2019

В пакете LTRCtrees можно приспособить дерево решений к специальному формату функции Surv: Surv (время, время2, событие), как показано в примере ниже 

set.seed(0)
library(survival)
library(LTRCtrees)
## Create the start-stop-event triplet needed for coxph and LTRC trees
first <- with(pbcseq, c(TRUE, diff(id) !=0)) #first id for each subject
last <- c(first[-1], TRUE) #last id
time1 <- with(pbcseq, ifelse(first, 0, day))
time2 <- with(pbcseq, ifelse(last, futime, c(day[-1], 0)))
event <- with(pbcseq, ifelse(last, status, 0))
event <- 1*(event==2)

pbcseq$time1 <- time1
pbcseq$time2 <- time2
pbcseq$event <-  event

## Fit the Cox model and LTRC trees with time-varying covariates
fit.cox <- coxph(Surv(time1, time2, event) ~ age + sex + log(bili), pbcseq)
LTRCIT.fit <- LTRCIT(Surv(time1, time2, event) ~ age + sex + log(bili), pbcseq)
LTRCART.fit <- LTRCART(Surv(time1, time2, event) ~ age + sex + log(bili), pbcseq)

Можно ли использовать ту же функцию в случайном лесу с помощью библиотеки RandomForestSRC 

library(randomForestSRC)
RF.fit <- rfsrc(Surv(time1, time2, event) ~ age + sex + log(bili), data=pbcseq, nsplit = 3, ntree = 100, importance = TRUE)

, который генерирует ошибку: 

Error in parseFormula(formula, data, ytry) : 
  Survival formula incorrectly specified.
...