A Comparative Study of Machine Learning Models for Heart Disease Prediction Using Grid Search and Random Search for Hyperparameter Tuning

Authors

  • Suman Arshad Deparment of Basic Sciences, Superior University, Lahore, 54000, Pakistan.
  • Syed Muhammad Junaid Zaidi Deparment of Basic Sciences, Superior University, Lahore, 54000, Pakistan.
  • Muazzam Ali Deparment of Basic Sciences, Superior University, Lahore, 54000, Pakistan.
  • M U Hashmi Deparment of Basic Sciences, Superior University, Lahore, 54000, Pakistan.
  • Abdul Manan Deparment of Basic Sciences, Superior University, Lahore, 54000, Pakistan.
  • Affan Ahmad Deparment of Computer Science, Superior University, Lahore, 54000, Pakistan.

Keywords:

Hyperparameters Modification, Random Forests, Machine Learning, Medical Testing and Diagnostics, Heartbeat Predicting, Ensembles Methods

Abstract

An important global health concern is heart failure, for which early detection can greatly improve patient outcomes. Machine learning has proved to be useful in predicting the likelihood of heart disease by looking at factors like age, high blood pressure, and cholesterol. This study compares popular machine learning models, such as Random Forests, Gradient Boosting, Stacking, KNN, SVM, and Logistic Regression. We utilized a Grid Search as well as Random Search to improve the models' efficiency and perform-ability. Following model tuning, the models were determined using metrics like accuracy, recall, F1 score, and precision, AUC, Cohen's Kappa, and MCC. With grid search accuracy of 94.95% and random search accuracy of 94.54%, Random Forest produced the best results. This highlights how important it is to select the right model and adjust its parameters for the best results, and it also shows how well Random Forest predicts heart disease.

Downloads

Published

2024-09-29

How to Cite

Suman Arshad, Syed Muhammad Junaid Zaidi, Muazzam Ali, M U Hashmi, Abdul Manan, & Affan Ahmad. (2024). A Comparative Study of Machine Learning Models for Heart Disease Prediction Using Grid Search and Random Search for Hyperparameter Tuning. Journal of Computing & Biomedical Informatics. Retrieved from https://jcbi.org/index.php/Main/article/view/697

Issue

Section

Articles