Predictive Modelling for Heart Disease Diagnosis: A Comparative Study of Classifiers

Authors

  • Nidhi Agarwal Indira Gandhi Delhi Technical University for Women image/svg+xml
  • Deepakshi Indira Gandhi Delhi Technical University for Women image/svg+xml
  • J Harikiran Vellore Institute of Technology University image/svg+xml
  • Yampati Bhagya Lakshmi Koneru Lakshmaiah Education Foundation image/svg+xml
  • Aylapogu Pramod Kumar Mallareddy Engineering College
  • Elangovan Muniyandy Saveetha Institute of Medical and Technical Sciences
  • Amit Verma Chandigarh University image/svg+xml

DOI:

https://doi.org/10.4108/eetpht.10.5518

Keywords:

Heart disease prediction, Machine learning classifiers, Naive Bayes, Logistic Regression, k-Nearest Neighbors

Abstract

INTRODUCTION: Cardiovascular diseases, including heart disease, remain a significant cause of morbidity and mortality worldwide. Timely and accurate diagnosis of heart disease is crucial for effective intervention and patient care. With the emergence of machine learning techniques, there is a growing interest in leveraging these methods to enhance diagnostic accuracy and predict disease outcomes.

OBJECTIVES: This study evaluates the performance of three machine learning classifiers—Naive Bayes, Logistic Regression, and k-Nearest Neighbors in predicting heart disease based on patient attributes.

METHODS: In this study, we explore the application of three prominent machine learning classifiers—Naive Bayes, Logistic Regression, and k-Nearest Neighbors (kNN)—to predict the presence of heart disease based on a set of patient attributes.

RESULTS: Using a dataset of 303 patient records with 14 attributes, including age, sex, and cholesterol levels, the data is pre-processed, scaled, and split into training and test sets. Each classifier is trained on the training set and evaluated on the test set. Results reveal that Naive Bayes and k-Nearest Neighbors classifiers outperform Logistic Regression in terms of accuracy, precision, recall, and area under the ROC curve (AUC).

CONCLUSION: This study underscores the promising role of machine learning in medical diagnosis, showcasing the potential of Naive Bayes and k-Nearest Neighbors classifiers in improving heart disease prediction accuracy. Future work could explore advanced classifiers and feature selection techniques to enhance predictive accuracy and generalize findings to larger datasets.

Downloads

Download data is not yet available.

References

Abdel-Basset, M. et al.: A novel group decision making model based on neutrosophic sets for heart disease diagnosis. Multimed Tools Appl. 79, 15–16, 9977–10002 (2020). https://doi.org/10.1007/S11042-019-07742-7/METRICS. DOI: https://doi.org/10.1007/s11042-019-07742-7

Agarwal, N: Virtual modem for STB network, 2011 2nd International Conference on Computer and Communication Technology (ICCCT-2011), Allahabad, 2011, pp. 108-112, doi: 10.1109/ICCCT.2011.6075126. DOI: https://doi.org/10.1109/ICCCT.2011.6075126

Abed, F.N. et al.: Enhancement Modelling Based on Electrical Discharge Machining Successive Discharges. Advances in Materials Science and Engineering. 2022, (2022). https://doi.org/10.1155/2022/8017375. DOI: https://doi.org/10.1155/2022/8017375

Agarwal, N. et al.: A Machine Learning Model to Prune Insignificant Attributes. In: 2021 9th International Conference on Reliability, Infocom Technologies and Optimization (Trends and Future Directions), ICRITO 2021. (2021). https://doi.org/10.1109/ICRITO51393.2021.9596425. DOI: https://doi.org/10.1109/ICRITO51393.2021.9596425

Agarwal, N. et al.: A Novel Fbit_sort Algorithm for digital encoding process. In: Proceedings - 2022 6th International Conference on Intelligent Computing and Control Systems, ICICCS 2022. (2022). https://doi.org/10.1109/ICICCS53718.2022.9788168. DOI: https://doi.org/10.1109/ICICCS53718.2022.9788168

Agarwal, N. et al.: A Novel Model to Predict the Effects of Enhanced Students’ Computer Interaction on Their Health in COVID-19 Pandemics. New Gener Comput. 1–34 (2023). https://doi.org/10.1007/S00354-023-00224-3/METRICS. DOI: https://doi.org/10.1007/s00354-023-00224-3

Agarwal, N. et al.: Applying XGBoost Machine Learning Model to Succor Astronomers Detect Exoplanets in Distant Galaxies. (2022). https://doi.org/10.1007/978-3-030-95711-7_33. DOI: https://doi.org/10.1007/978-3-030-95711-7_33

Agarwal, N. et al.: Exigency of Natural Language processing to skirmish with COVID-19 pandemic. International Journal of Advances in Engineering and Management (IJAEM). 3, 1081 (2021). https://doi.org/10.35629/5252-030710811086.

Agarwal, N. et al.: Intelligent Aquila Optimization Algorithm-Based Node Localization Scheme for Wireless Sensor Networks. Computers, Materials and Continua. 74, 1, (2023). https://doi.org/10.32604/cmc.2023.030074. DOI: https://doi.org/10.32604/cmc.2023.030074

Gupta A, Vardhan H, Varshney S, Saxena S, Singh S, Agarwal N. Kconnect:The Design and Development of Versatile Web Portal for Enhanced Collaboration and Communication. EAI Endorsed Scal Inf Syst

Agarwal, N. et al.: Multiclass Classification of Different Glass Types using Random Forest Classifier. In: Proceedings - 2022 6th International Conference on Intelligent Computing and Control Systems, ICICCS 2022. (2022). https://doi.org/10.1109/ICICCS53718.2022.9788326. DOI: https://doi.org/10.1109/ICICCS53718.2022.9788326

Agarwal, N. et al.: Semi-Supervised Learning with GANs for Melanoma Detection. In: Proceedings - 2022 6th International Conference on Intelligent Computing and Control Systems, ICICCS 2022. (2022). https://doi.org/10.1109/ICICCS53718.2022.9787990. DOI: https://doi.org/10.1109/ICICCS53718.2022.9787990

Agarwal, N., Deep, P.: Obtaining better software product by using test first programming technique. In: Proceedings of the 5th International Conference on Confluence 2014: The Next Generation Information Technology Summit. (2014). https://doi.org/10.1109/CONFLUENCE.2014.6949233.

Agarwal, N., Tayal, D.K.: A new model based on the extended COPRAS method for improving performance during the accreditation process of Indian Higher Educational Institutions. Computer Applications in Engineering Education. 31, 3, (2023). https://doi.org/10.1002/cae.22602. DOI: https://doi.org/10.1002/cae.22602

Agarwal, N., Tayal, D.K.: A Novel Model to Predict the Whack of Pandemics on the International Rankings of Academia. 24–38 (2023). https://doi.org/10.1007/978-3-031-35081-8_3. DOI: https://doi.org/10.1007/978-3-031-35081-8_3

Agarwal, N., Tayal, D.K.: FFT based ensembled model to predict ranks of higher educational institutions. Multimed Tools Appl. 81, 23, (2022). https://doi.org/10.1007/s11042-022-13180-9. DOI: https://doi.org/10.1007/s11042-022-13180-9

N. Agarwal and P. Deep, "Obtaining better software product by using test first programming technique," 2014 5th International Conference - Confluence The Next Generation Information Technology Summit (Confluence), Noida, India, 2014, pp. 742-747, doi: 10.1109/CONFLUENCE.2014.6949233. DOI: https://doi.org/10.1109/CONFLUENCE.2014.6949233

Anushka et al.: Predicting Credit Card Defaults with Machine Learning Algorithm Using Customer Database. 262–277 (2023). https://doi.org/10.1007/978-3-031-35078-8_22. DOI: https://doi.org/10.1007/978-3-031-35078-8_22

Jha, A. et al.: Movie Recommendation Using Content-Based and Collaborative Filtering Approach. Lecture Notes of the Institute for Computer Sciences, Social-Informatics and Telecommunications Engineering, LNICST. 470 LNICST, 439–450 (2023). https://doi.org/10.1007/978-3-031-35078-8_37/COVER. DOI: https://doi.org/10.1007/978-3-031-35078-8_37

Agarwal N, Arora I, Saini H, Sharma U. A Novel Approach for Earthquake Prediction Using Random Forest and Neural Networks. EAI Endorsed Trans Energy Web.

Kimenai, D.M. et al.: Sex Differences in Cardiac Troponin I and T and the Prediction of Cardiovascular Events in the General Population. https://doi.org/10.1093/clinchem/hvab109. DOI: https://doi.org/10.1093/clinchem/hvab109

Mohan, S. et al.: Effective Heart Disease Prediction Using Hybrid Machine Learning Techniques. IEEE Access. 7, 81542–81554 (2019). https://doi.org/10.1109/ACCESS.2019.2923707. DOI: https://doi.org/10.1109/ACCESS.2019.2923707

Shanmukha, R.K.N.S., Thinakaran, K.: Prediction of Heart Disease using Decision Tree over Logistic Regression using Machine Learning with Improved Accuracy. CARDIOMETRY. 25, 1514–1519 (2023). https://doi.org/10.18137/CARDIOMETRY.2022.25.15141519. DOI: https://doi.org/10.18137/cardiometry.2022.25.15141519

Sharma, T. et al.: Machine Learning/Deep Learning for Natural Disasters. Applications of Artificial Intelligence, Big Data and Internet of Things in Sustainable Development. 91–121 (2022). https://doi.org/10.1201/9781003245469-7. DOI: https://doi.org/10.1201/9781003245469-7

Sharma, V. et al.: Heart Disease Prediction using Machine Learning Techniques. Proceedings - IEEE 2020 2nd International Conference on Advances in Computing, Communication Control and Networking, ICACCCN 2020. 177–181 (2020). https://doi.org/10.1109/ICACCCN51052.2020.9362842. DOI: https://doi.org/10.1109/ICACCCN51052.2020.9362842

Singh, A., Kumar, R.: Heart Disease Prediction Using Machine Learning Algorithms. 2020 International Conference on Electrical and Electronics Engineering (ICE3). 452–457 (2020). https://doi.org/10.1109/ICE348803.2020.9122958. DOI: https://doi.org/10.1109/ICE348803.2020.9122958

Sivaparthipan, C.B. et al.: IoT-based patient stretcher movement simulation in smart hospital using type-2 fuzzy sets systems. Production Planning and Control. (2023). https://doi.org/10.1080/09537287.2023.2217419. DOI: https://doi.org/10.1080/09537287.2023.2217419

Srivastav, A. et al.: A Novel Fuzzy Graph Connectivity Measure to Perform Word Sense Disambiguation Using Fuzzy Hindi WordNet. In: 3rd IEEE 2022 International Conference on Computing, Communication, and Intelligent Systems, ICCCIS 2022. (2022). https://doi.org/10.1109/ICCCIS56430.2022.10037658. DOI: https://doi.org/10.1109/ICCCIS56430.2022.10037658

Tayal, D.K. et al.: To Predict the Fire Outbreak in Australia using Historical Database. In: 2022 10th International Conference on Reliability, Infocom Technologies and Optimization (Trends and Future Directions), ICRITO 2022. (2022). https://doi.org/10.1109/ICRITO56286.2022.9964603. DOI: https://doi.org/10.1109/ICRITO56286.2022.9964603

Downloads

Published

22-03-2024

How to Cite

1.
Agarwal N, Deepakshi, Harikiran J, Lakshmi YB, Pramod Kumar A, Muniyandy E, Verma A. Predictive Modelling for Heart Disease Diagnosis: A Comparative Study of Classifiers. EAI Endorsed Trans Perv Health Tech [Internet]. 2024 Mar. 22 [cited 2024 Nov. 10];10. Available from: https://publications.eai.eu/index.php/phat/article/view/5518