之前只画过分类树的roc曲线,用的predict函数里的prob类型,但是换生存树就一脸懵逼,查了好久资料都没找到ROC怎么画
拟合生存树的代码如下:
- library(survival) #用以拟合生存函数
- library("rpart") #构建生存树
- library("partykit") #以标准化的方式绘制生存树
- library(rattle)
- library(rpart.plot)
- temp = coxph(Surv(pgtime, pgstat) ~ 1, stagec)
- newtime = predict(temp, type = 'expected')
- fit <- rpart(Surv(pgtime, pgstat) ~ ., data = stagec)
- cp=fit$cptable[which.min(fit$cptable[,"xerror"]),"CP"] #剪枝
- pfit <- prune(fit, cp)
- fancyRpartPlot(pfit)
- tfit <- as.party(pfit) #转换成party类型
- predtree<-predict(tfit,newdata=stagec,type="prob") #预测