Open Access iconOpen Access

ARTICLE

crossmark

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: email

Intelligent Automation & Soft Computing 2021, 30(3), 929-943. https://doi.org/10.32604/iasc.2021.018382

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

APA Style
Dubey, A.K., Choudhary, K., Sharma, R. (2021). Predicting heart disease based on influential features with machine learning. Intelligent Automation & Soft Computing, 30(3), 929-943. https://doi.org/10.32604/iasc.2021.018382
Vancouver Style
Dubey AK, Choudhary K, Sharma R. Predicting heart disease based on influential features with machine learning. Intell Automat Soft Comput . 2021;30(3):929-943 https://doi.org/10.32604/iasc.2021.018382
IEEE Style
A.K. Dubey, K. Choudhary, and R. Sharma "Predicting Heart Disease Based on Influential Features with Machine Learning," Intell. Automat. Soft Comput. , vol. 30, no. 3, pp. 929-943. 2021. https://doi.org/10.32604/iasc.2021.018382



cc 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.
  • 1669

    View

  • 1375

    Download

  • 0

    Like

Share Link