An empirical study on machine learning algorithms for heart disease prediction

• Published in: IAES International Journal of Artificial Intelligence Vol. 11; no. 3; p. 1066
• Main Authors: Assegie, Tsehay Admassu, Rangarajan, Prasanna Kumar, Kumar, Napa Komal, Vigneswari, Dhamodaran
• Format: Journal Article
• Language: English
• Published: Yogyakarta IAES Institute of Advanced Engineering and Science 01.09.2022
• Subjects: Accuracy; Algorithms; Cardiovascular disease; Datasets; Decision trees; Experimentation; Heart; Heart diseases; Machine learning; Prediction models; Support vector machines
• ISSN: 2089-4872; 2252-8938; 2089-4872
• DOI: 10.11591/ijai.v11.i3.pp1066-1073

In recent years, machine learning is attaining higher precision and accuracy in clinical heart disease dataset classification. However, literature shows that the quality of heart disease feature used for the training model has a significant impact on the outcome of the predictive model. Thus, this study focuses on exploring the impact of the quality of heart disease features on the performance of the machine learning model on heart disease prediction by employing recursive feature elimination with cross-validation (RFECV). Furthermore, the study explores heart disease features with a significant effect on model output. The dataset for experimentation is obtained from the University of California Irvine (UCI) machine learning dataset. The experiment is implemented using a support vector machine (SVM), logistic regression (LR), decision tree (DT), and random forest (RF) are employed. The performance of the SVM, LR, DT, and RF models. The result appears to prove that the quality of the feature significantly affects the performance of the model. Overall, the experiment proves that RF outperforms as compared to other algorithms. In conclusion, the predictive accuracy of 99.7% is achieved with RF.