基于机器学习方法的原发性肝癌患者预后预测研究

摘要

目的:使用机器学习方法对原发性肝癌患者生存和复发时间进行预测,并比较不同机器学习方法的预测效果.方法:使用Logistic回归、随机森林、SVM、C5.0决策树、神经网络、Bagging算法和Adaboost算法构建患者肝切除手术后3年无瘤生存时间和3年总生存时间的预后预测模型.结果:通过Logistic回归模型找到了影响患者生存和复发的影响因素,并且支持向量机和随机森林有较好的效果,测试集上的AUC为0.7122和0.7103.结论:机器学习方法相较于Logistic回归模型有较好的预测效果,但是Logistic回归模型有较好的可解释性.在分析实际问题时,可以采用二者结合的方法.