Comparing some Machine Learning Models for Cardiovascular Disease

Sara Noori Mohammad Ali; Nawzad Muhammed Ahmed

doi:10.47310/jpms2025140408

Author(s)

Sara Noori Mohammad Ali Department of Statistics and Informatics, College of Administration and Economics, University of Sulaimani- Sulaymaniyah, Iraq

Nawzad Muhammed Ahmed Department of Statistics and Informatics, College of Administration and Economics, University of Sulaimani- Sulaymaniyah, Iraq

Abstract

Background and Aim: Cardiovascular disease remains a leading cause of morbidity and mortality worldwide, necessitating the development of accurate predictive models for early diagnosis. Therefore, this study aimed to evaluate and compare the performance of three machine learning models-Random Forest, Decision Tree, and K-Nearest Neighbors-in predicting cardiovascular disease based on key risk factors. Method: This retrospective study utilized patient data from Shar Hospital in Sulaimaniyah City. The dataset included demographic and clinical risk factors such as age, smoking status, diabetes, hypertension, and family history of cardiovascular disease. The three machine learning models were trained and tested using various data-splitting ratios, and their performance was assessed using accuracy, F1-score, recall, precision, and specificity. Statistical analysis and model validation were conducted using Python in Jupyter Notebook. Results: A total of 300 patient records were included in the study. The Random Forest model demonstrated the highest predictive accuracy compared to Decision Tree and K-Nearest Neighbors, consistently outperforming the other models across different training-testing configurations. Feature importance analysis revealed that age and family history were the most influential predictors of cardiovascular disease, whereas gender and marital status had minimal impact. The confusion matrix further confirmed the reliability of the Random Forest model, showing a high number of correctly classified cases with minimal false positives and false negatives. Conclusions: The findings indicate that Random Forest is the most effective model for cardiovascular disease prediction, with strong classification performance and high accuracy. The study also highlights the importance of age and family history as dominant risk factors. These results support the application of machine learning in clinical settings for early detection and risk assessment, enabling better-informed medical interventions.

Publication Details

Page(s) 60-66

DOI 10.47310/jpms2025140408

Published Journal: Journal of Pioneering Medical Sciences, Volume: 14, Issue: 4, Year: 2025

Request the full text from PASTIC

Keywords

Machine Learning Models, Medical data classification, Cardiovascular Disease Prediction, Risk Factors Analysis

References

Nowbar Alexandra N. . .Mortality from ischemic heart disease: Analysis of data from the World Health Organization and coronary artery disease risk factors From NCD Risk Factor Collaboration. Circulation: Cardiovascular Quality and Outcomes, 12(6) : .

Ogunpola ,Adedayo ,January .2024 .Machine learning-based predictive models for detection of cardiovascular diseases. ” Diagnostics, 14(2) : .

Mirza ,Bilal ,January .2019 .Machine learning and integrative analysis of biomedical big data. ” Genes, 10(2) : .

Halabaku ,January .2019 .? direct=true&profile=ehost&scope=site&authtype=crawler&jr nl=10798587&AN=182059750&h=VYgprPLJJREWnmXtD AknI55aYvFvoaRXsipPD04oL0gRF53FJ73F506TCevv0ngO PDjBD%2FW4TBg1Og10KvHwzQ%3D%3D&crl=c.. ” Intelligent Automation & Soft Computing, 39(6) : .

Priyanka and Dharmender,Kumar and Dharmender,April 2020 . .“Decision tree classifier: A detailed survey. International Journal of Information and Decision Sciences, 020(3) : 246-269.

Mohan ,Senthilkumar . .Effective heart disease prediction using hybrid machine learning techniques. ” IEEE access, 7 : 81542-81554.

Pathan Muhammad,Salman Muhammad,November Muhammad .2022 .Analyzing the impact of feature selection on the accuracy of heart disease prediction. ” Healthcare Analytics, 2 : .

Bashir ,Saba .2022 .A novel feature selection method for classification of medical data using filters, wrappers, and embedded approaches. ” Complexity, 1 : .

Parisi ,Rosa .2020 .National, regional, and worldwide epidemiology of psoriasis: systematic analysis and modelling study. ” BMJ, 369 : .

Dhiman ,Paula ,January .2014 .Availability and quality of coronary heart disease family history in primary care medical records: implications for cardiovascular risk assessment. PLoS One, 0081998(1) : .