libiary(survival)
library(survivalROC)
fit_curve_train4<-survfit(Surv(life_span,status)~risk,data = risk_train4data)
plot(fit_curve_train4,,col=c("red","blue"),xlab = "time(years)",ylab = "survival rate",
main="survival curve of train set",mark.time=T)
legend(5,.4,paste("High risk (n=",nrow(risk_train4data[risk_train4data[,5]=="high",]),")",sep = ""),lty = NULL,text.col = "red",bty = "n")
legend(5,.3,paste("Low risk (n=",nrow(risk_train4data[risk_train4data[,5]=="low",]),")",sep = ""),lty = NULL,text.col = "blue",bty = "n")
legend(15,.2,paste("P value =",p_value_train4,sep = ""),lty = NULL,col = "black",bty = "n")
fit_curve_test4<-survfit(Surv(life_span,status)~risk,data = risk_test4data)
plot(fit_curve_test4,,col=c("red","blue"),xlab = "time(years)",ylab = "survival rate",
main="survival curve of test set",mark.time=T)
legend(5,.4,paste("High risk (n=",nrow(risk_test4data[risk_test4data[,5]=="high",]),")",sep = ""),lty = NULL,text.col = "red",bty = "n")
legend(5,.3,paste("Low risk (n=",nrow(risk_test4data[risk_test4data[,5]=="low",]),")",sep = ""),lty = NULL,text.col = "blue",bty = "n")
legend(15,.2,paste("P value =",p_value_test4,sep = ""),lty = NULL,col = "black",bty = "n")
fit_curve_all_set4<-survfit(Surv(life_span,status)~risk,data = all_set4)
plot(fit_curve_all_set4,col=c("red","blue"),xlab = "time(years)",ylab = "survival rate",
main="survival curve of all set",mark.time=T)
legend(5,.4,paste("High risk (n=",nrow(all_set4[all_set4[,5]=="high",]),")",sep = ""),lty = NULL,text.col = "red",bty = "n")
legend(5,.3,paste("Low risk (n=",nrow(all_set4[all_set4[,5]=="low",]),")",sep = ""),lty = NULL,text.col = "blue",bty = "n")
legend(15,.2,paste("P value =",p_value_all4,sep = ""),lty = NULL,col = "black",bty = "n")
##绘制ROC 曲线
roc<-survivalROC(Stime = risk_train4data$life_span,status = risk_train4data$status,marker = risk_train4data$riskscore,
predict.time = 5,method = "KM")
plot(roc$FP,roc$TP,type = "l",xlim = c(0,1),ylim = c(0,1),col="green")
roc$AUC
roctest4<-survivalROC(Stime = risk_test4data$life_span,status = risk_test4data$status,marker = risk_test4data$riskscore,
predict.time = 5,method = "KM")
plot(roctest4$FP,roctest4$TP,type = "l",xlim = c(0,1),ylim = c(0,1),col="green")
rocall4<-survivalROC(Stime = all_set4$life_span,status = all_set4$status,marker = all_set4$riskscore,
predict.time = 5,method = "KM")
plot(rocall4$FP,rocall4$TP,type = "l",xlim = c(0,1),ylim = c(0,1),col="green")
