Predicting Heart Disease Risk Using Machine Learning
DOI:
https://doi.org/10.61173/nf9q0t31Keywords:
Heart disease prediction, Machine learning, Logistic regression, Random forest, XGBoostAbstract
Heart disease is considered the world leader in mortality rates among people. A project was created with the goal of deploying a machine learning model for the prediction of cardiac disease using publicly available datasets from the University of California, Irvine Machine Learning Repository. Early detection through prompt care may improve death rates. In this respect, several machine learning algorithms: decision trees, logistic regression, random forests, and XGBoost, were used to identify heart disease patterns and risk factors. These above-mentioned models will be evaluated against the following key performance metrics: precision, accuracy, recall, and F1-score. Of all the algorithms, the XGBoost model performed the best, giving a precision of 89% and an F1-score of 0.87, which was one of the best in predicting heart diseases. These findings emphasize the crucial role of machine learning in further improving the prediction of cardiovascular diseases, possibly allowing for early diagnosis. Such predictive tools will allow healthcare providers to move toward more personalized and preventive treatments in patient care and outcomes.References
[1] World Health Organization (WHO). Cardiovascular diseases (CVDs). 2021.
[2] Yusuf S, Joseph P, Rangarajan S, Islam S, Mente A, Hystad P, Teo K. Modifiable risk factors, cardiovascular disease, and mortality in 155,722 individuals from 21 high-income, middleincome, and low-income countries (PURE): a prospective cohort study. The Lancet, 2020, 395(10226): 795-808.
[3] Sharma V, Yadav S, Gupta M. Heart disease prediction using machine learning techniques. In 2020 2nd international conference on advances in computing, communication control and networking (ICACCCN). IEEE, 2020: 177-181.
[4] Aljanabi M, Qutqut M H, Hijjawi M. Machine learning classification techniques for heart disease prediction: a review. International Journal of Engineering & Technology, 2018, 7(4): 5373-5379.
[5] Patel P, Chaudhary A, Bhatnagar S. A review of machine learning algorithms for heart disease prediction. Algorithms, 2023, 16(2): 88.
[6] Mehrabani-Zeinabad H, Pourmohammadi K, Rahimi M. Predictive modeling of heart disease using machine learning algorithms. BMC Medical Informatics and Decision Making, 2023, 23: 21.
[7] Podkowinski D, Beka S, Mursch-Edlmayr A S, et al. A swept source optical coherence tomography angiography study: imaging artifacts and comparison of non-perfusion areas with fluorescein angiography in diabetic macular edema. PLoS One, 2021, 16(4): e0249918.
[8] Jindal H, Agrawal S, Khera R, Jain R, Nagrath P. Heart disease prediction using machine learning algorithms. IOP Conference Series: Materials Science and Engineering, 2021, 1022(1): 012072.
[9] Fitriyani N L, Syafrudin M, Alfian G, Rhee J. HDPM: an effective heart disease prediction model for a clinical decision support system. IEEE Access, 2020, 8: 133034-133050.
[10] Ayon S I, Islam M M, Hossain M R. Coronary artery heart disease prediction: a comparative study of computational intelligence techniques. IETE Journal of Research, 2022, 68(4): 2488-2507.
[11] Kavitha M, Gnaneswar G, Dinesh R, Sai Y R, Suraj R S. Heart disease prediction using hybrid machine learning model. 2021 6th international conference on inventive computation technologies (ICICT). IEEE, 2021: 1329-1333.
[12] UCI Machine Learning Repository. Heart disease dataset.
Downloads
Published
Issue
Section
License
Copyright (c) 2024 by the authors.

This work is licensed under a Creative Commons Attribution 4.0 International License.
