RESEARCH ARTICLE


Grid Search based Optimum Feature Selection by Tuning hyperparameters for Heart Disease Diagnosis in Machine learning



G. Saranya1, *, A. Pravin2
1 Department of Networking and Communications, SRM Institute of Science and Technology, Kattankulathur, Chennai, Tamil Nadu, India
2 Department of Computer Science and Engineering, Sathyabama Institute of Science and Technology, Tamil Nadu, India


Article Metrics

CrossRef Citations:
0
Total Statistics:

Full-Text HTML Views: 713
Abstract HTML Views: 249
PDF Downloads: 228
ePub Downloads: 172
Total Views/Downloads: 1362
Unique Statistics:

Full-Text HTML Views: 506
Abstract HTML Views: 158
PDF Downloads: 173
ePub Downloads: 127
Total Views/Downloads: 964



Creative Commons License
© 2023 Saranya and Pravin

open-access license: This is an open access article distributed under the terms of the Creative Commons Attribution 4.0 International Public License (CC-BY 4.0), a copy of which is available at: https://creativecommons.org/licenses/by/4.0/legalcode. This license permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.

* Address correspondence to this author at the Department of Networking and Communications, SRM Institute of Science and Technology, Kattankulathur, Chennai, Tamil Nadu, India; E-mail: saranyag3@srmist.edu.in


Abstract

Background:

Heart disease prediction model helps physicians to identify patients who are at high risk of developing heart disease and target prevention strategies accordingly. These models use patient demographics, medical history, lifecycle factors, and clinical measurements to calculate the risk of heart disease within a certain time frame. In identifying important features of heart disease, a popular approach is using Machine learning (ML) models. ML models can analyse a large amount of data and find patterns that are difficult for humans to detect.

Methods:

In this proposed work, Random Forest classifier is used to identify the most important features that contribute to heart disease and increase the prediction accuracy of the model by tuning the hyperparameters using grid search approach.

Results:

The proposed system was evaluated and compared in terms of accuracy, error rate and recall with the traditional system. As the traditional system achieved accuracies between 81.97% and 90.16%., the proposed hyperparameter tuning model achieved accuracies in the range increased between 84.22% and 96.53%.

Conclusion:

These evaluations demonstrated that the proposed prediction approach is capable of achieving more accurate results compared with the traditional approach in predicting heart disease by finding optimum features.

Keywords: Grid search optimization (GSO), Machine learning(ML), Random forest (RF), Hyperparameter tuning, Heart disease, Feature selection (FS).