Vol.30, No.3, 2021, pp.929-943, doi:10.32604/iasc.2021.018382
OPEN ACCESS
ARTICLE
Predicting Heart Disease Based on Influential Features with Machine Learning
  • Animesh Kumar Dubey*, Kavita Choudhary, Richa Sharma
Institute of Engineering and Technology, JK Lakshmipat University, Jaipur, 302026, India
* Corresponding Author: Animesh Kumar Dubey. Email:
Received 06 March 2021; Accepted 11 May 2021; Issue published 20 August 2021
Abstract
Heart disease is a major health concern worldwide. The chances of recovery are bright if it is detected at an early stage. The present report discusses a comparative approach to the classification of heart disease data using machine learning (ML) algorithms and linear regression and classification methods, including logistic regression (LR), decision tree (DT), random forest (RF), support vector machine (SVM), SVM with grid search (SVMG), k-nearest neighbor (KNN), and naive Bayes (NB). The ANOVA F-test feature selection (AFS) method was used to select influential features. For experimentation, two standard benchmark datasets of heart diseases, Cleveland and Statlog, were obtained from the UCI Machine Learning Repository. The performance of the machine learning models was examined for accuracy, precision, recall, F-score, and Matthews correlation coefficient (MCC), along with error rates. The results indicated that RF and SVM with grid search algorithms performed better on the Cleveland dataset, while the LR and NB classifiers performed better on the Statlog dataset. Outcomes improved significantly when classification was performed after applying AFS, except for NB, for both datasets.
Keywords
LR; DT; RF; KNN; SVM
Cite This Article
A. Kumar Dubey, K. Choudhary and R. Sharma, "Predicting heart disease based on influential features with machine learning," Intelligent Automation & Soft Computing, vol. 30, no.3, pp. 929–943, 2021.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.