Indonesian Journal of Electrical Engineering and Computer Science
Vol 21, No 3: March 2021

Predicting heart failure using a wrapper-based feature selection

Minh Tuan Le (Ho Chi Minh City)
Minh Thanh Vo (Ho Chi Minh City)
Nhat Tan Pham (Ho Chi Minh City)
Son V.T Dao (Ho Chi Minh City)



Article Info

Publish Date
01 Mar 2021

Abstract

In the current health system, it is very difficult for medical practitioners/physicians to diagnose the effectiveness of heart contraction. In this research, we proposed a machine learning model to predict heart contraction using an artificial neural network (ANN). We also proposed a novel wrapper-based feature selection utilizing a grey wolf optimization (GWO) to reduce the number of required input attributes. In this work, we compared the results achieved using our method and several conventional machine learning algorithms approaches such as support vector machine, decision tree, K-nearest neighbor, naïve bayes, random forest, and logistic regression. Computational results show not only that much fewer features are needed, but also higher prediction accuracy can be achieved around 87%. This work has the potential to be applicable to clinical practice and become a supporting tool for doctors/physicians.

Copyrights © 2021