Heart failure prediction model based on machine learning algorithms
10.16352/j.issn.1001-6325.2024.06.0845
- VernacularTitle:基于机器学习算法的心力衰竭预测模型
- Author:
Chuanli HU
1
;
Xiaosong HE
;
Jiang ZHAO
;
Hua LI
Author Information
1. 陆军军医大学第二附属医院 麻醉科 重庆 400037
- Keywords:
heart failure;
machine learning;
prediction
- From:
Basic & Clinical Medicine
2024;44(6):845-852
- CountryChina
- Language:Chinese
-
Abstract:
Objective To construct a model of heart failure risk prediction based on four machine learning algo-rithms in order to support early diagnosis and intervention.Methods After reviewing the heart failure dataset pub-lished on the Kaggle community,feature selection was used to select relevant factors related to heart failure as pre-dictive indicators.Four machine learning algorithms,namely logistic regression,support vector machine,random forest,and XGBoost were selected to establish predictive models.Compared and analyzed its accuracy,precision,recall,F1 score and area under the ROC curve(AUC)to verify the performance of the model.Results The study analyzed 11 features of 918 patients with heart failure and selected 10 feature factors for modeling.After optimizing the hyper-parameters through grid search,the XGBoost model performed the best,with accuracy,precision,recall,and f1_score and AUC values were 87.5%,90.38%,89.71%,90.04%and 0.93,respectively.In addition,data analysis showed that exercise ST slope,chest pain type,and exercise induced angina were main influencing factors for heart failure.Conclusions The XG Boost model has the best predictive tool for heart failure,and machine learning algorithms may support early prevention,early diagnosis as well as control of heart failure.
