基于Bayes超参数优化梯度提升树的心脏病预测方法

王海燕, 焦增晨, 赵剑, 安天博, 鞠熠

PDF(1484 KB)
PDF(1484 KB)
吉林大学学报(理学版) ›› 2025, Vol. 63 ›› Issue (02) : 472-478. DOI: 10.13413/j.cnki.jdxblxb.2024223

基于Bayes超参数优化梯度提升树的心脏病预测方法

  • 王海燕, 焦增晨, 赵剑, 安天博, 鞠熠
作者信息 +
History +

摘要

针对传统机器学习算法在数据集Cleveland和Hungary上预测准确率低的问题,提出一种基于Bayes超参数优化梯度提升树的心脏病预测方法.首先,采用K-最近邻算法对数据集中的缺失值进行填补,用Min-Max标准化、 One-Hot编码处理数据,并基于梯度提升树算法进行心脏病预测;其次,采用Bayes优化和十倍交叉验证的方式搜寻算法的最佳超参数组合.实验结果表明,优化后的梯度提升树算法在心脏病数据集Cleveland上预测准确率可达90.2%,在心脏病数据集Hungary上预测准确率可达81.4%,优于决策树、支持向量机、K-最近邻等传统机器学习方法,可辅助医生进行心脏病诊断.

关键词

心脏病预测 / K-最近邻算法 / 梯度提升树 / Bayes优化

中图分类号

R541 / TP181

引用本文

导出引用
王海燕, 焦增晨, 赵剑, 安天博, 鞠熠. 基于Bayes超参数优化梯度提升树的心脏病预测方法. 吉林大学学报(理学版). 2025, 63(02): 472-478 https://doi.org/10.13413/j.cnki.jdxblxb.2024223

基金

吉林省教育厅科学技术研究项目(批准号:JJKH20220597KJ); 吉林省科技发展计划项目(批准号:YDZJ202201ZYTS549)

评论

PDF(1484 KB)

Accesses

Citation

Detail

段落导航
相关文章

/