Predictive Modelling for Heart Disease Diagnosis: A Comparative Study of Classifiers
DOI:
https://doi.org/10.4108/eetpht.10.5518Keywords:
Heart disease prediction, Machine learning classifiers, Naive Bayes, Logistic Regression, k-Nearest NeighborsAbstract
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
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
How to Cite
Issue
Section
License
Copyright (c) 2024 Nidhi Agarwal, Deepakshi, J Harikiran, Yampati Bhagya Lakshmi, Aylapogu Pramod Kumar, Elangovan Muniyandy, Amit Verma
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
This is an open access article distributed under the terms of the CC BY-NC-SA 4.0, which permits copying, redistributing, remixing, transformation, and building upon the material in any medium so long as the original work is properly cited.
