IAES International Journal of Artificial Intelligence (IJ-AI)
Vol 13, No 4: December 2024

Accuracy based-stacked ensemble learning model for the prediction of coronary heart disease

Bhutia, Santosini (Unknown)
Patra, Bichitrananda (Unknown)
Ray, Mitrabinda (Unknown)



Article Info

Publish Date
01 Dec 2024

Abstract

Coronary heart disease (CHD) is the primary cause of silent and noncommunicable deaths. Early detection is essential for slowing the progression of death and saving lives. Medical researchers use machine learning techniques to predict CHD. This article proposes an accuracy based-stacked ensemble learning (AB-SEL) model to predict CHD while minimizing computational time (CT). The dataset undergoes the logistic regression recursive feature elimination (LR-RFE) method to identify the important features. The three strong classifiers, logistic regression (LR), random forest (RF), and AdaBoost, are chosen to build ensemble machine-learning models, including techniques like bagging, majority voting, and stacking, for the Cleveland dataset accessible from Kaggle. Data scaling was done using the normal scalar method, and hyperparameter optimization was done using random search and grid search. Effectiveness is measured by accuracy, precision, recall, F1 score, and CT is validated through 5-fold cross-validation. The suggested approach achieved a 90.16% accuracy rate, required only 0.2 seconds of CT, and yielded an area under the curve (AUC) of 0.892.

Copyrights © 2024






Journal Info

Abbrev

IJAI

Publisher

Subject

Computer Science & IT Engineering

Description

IAES International Journal of Artificial Intelligence (IJ-AI) publishes articles in the field of artificial intelligence (AI). The scope covers all artificial intelligence area and its application in the following topics: neural networks; fuzzy logic; simulated biological evolution algorithms (like ...