The Prediction of Heart Failure based on Four Machine Learning Algorithms
DOI:
https://doi.org/10.54097/hset.v39i.6771Keywords:
Heart Disease; Decision Tree (DT); K-Nearest Neighbor (KNN); Support Vector Machine (SVM); Extreme Gradient Boosting (XGB).Abstract
The main purpose of this study is to observe the correlations between clinical features (input) with heart disease (output) in the practice of machine learning and modeling. Primarily, this research conducts a data exploration of the attributes and the output to observe the relationships between the attributes and heart disease. The experiment method covers the Decision Tree, K-Nearest Neighbor, Support Vector Machine and Extreme Gradient Boosting algorithms after the data exploration. This research concludes that exercise-induced angina and ST depression are two factors that are highly related to heart disease. Older males should take more care of these features because they are more likely to have heart disease. Moreover, based on the output of four algorithms, the SVM is the best method for predicting the probability of the occurrence of heart disease since SVM output the highest values on the accuracy, recall rate, and f1-score.
Downloads
References
“Heart Disease.” MedlinePlus, U.S. National Library of Medicine, (2022, September 2).
“Heart Disease Facts.” Centers for Disease Control and Prevention, Centers for Disease Control and Prevention, <https://www.cdc.gov/heartdisease/facts.htm> (2022, July 15).
Mohan, Senthilkumar, Chandrasegar Thirumalai, and Gautam Srivastava. "Effective heart disease prediction using hybrid machine learning techniques." IEEE access 7 (2019): 81542-81554.
Fitriyani, Norma Latif, et al. "HDPM: an effective heart disease prediction model for a clinical decision support system." IEEE Access 8 (2020): 133034-133050.
Garg, Apurv, Bhartendu Sharma, and Rijwan Khan. "Heart disease prediction using machine learning techniques." IOP Conference Series: Materials Science and Engineering. Vol. 1022. No. 1. IOP Publishing, 2021.
fedesoriano. (September 2021). Heart Failure Prediction Dataset. Retrieved (2022, August 23) from (2021, September 22).
Members, Writing Group, et al. "Executive Summary: Heart Disease and Stroke Statistics--2016 Update: A Report from the American Heart Association." Circulation 133.4 (2016): 447-454.
Virani, Salim S., et al. "heart disease and stroke statistics--2021 update: a report from the American Heart Association." Circulation 143.8 (2021): e254-e743.
Otchere, Daniel Asante, et al. "Application of supervised machine learning paradigms in the prediction of petroleum reservoir properties: Comparative analysis of ANN and SVM models." Journal of Petroleum Science and Engineering 200 (2021): 108182.
Tipping, Michael E. "Sparse Bayesian Learning and the Relevance Vector Machine." Journal of Machine Learning Research 1.3 (2001): 211-44. Web.
Vijayarani, S., and S. Dhayanand. "Liver disease prediction using SVM and Naïve Bayes algorithms." International Journal of Science, Engineering and Technology Research (IJSETR) 4.4 (2015): 816-8201.
Li, Hua, et al. "XGBoost model and its application to personal credit evaluation." IEEE Intelligent Systems 35.3 (2020): 52-61.
HUANG, Jingwei, et al. "An XGB-based runtime prediction algorithm for cloud workflow tasks."
Downloads
Published
Issue
Section
License
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
