Open Access iconOpen Access

ARTICLE

Machine Learning and Artificial Neural Network for Predicting Heart Failure Risk

Polin Rahman1, Ahmed Rifat1, MD. IftehadAmjad Chy1, Mohammad Monirujjaman Khan1,*, Mehedi Masud2, Sultan Aljahdali2

1 Department of Electrical and Computer Engineering, North South University, Bashundhara, Dhaka, 1229, Bangladesh
2 Department of Computer Science, College of Computers and Information Technology, Taif University, P. O. Box 11099, Taif, 21944, Saudi Arabia

* Corresponding Author: Mohammad Monirujjaman Khan. Email: email

Computer Systems Science and Engineering 2023, 44(1), 757-775. https://doi.org/10.32604/csse.2023.021469

Abstract

Heart failure is now widely spread throughout the world. Heart disease affects approximately 48% of the population. It is too expensive and also difficult to cure the disease. This research paper represents machine learning models to predict heart failure. The fundamental concept is to compare the correctness of various Machine Learning (ML) algorithms and boost algorithms to improve models’ accuracy for prediction. Some supervised algorithms like K-Nearest Neighbor (KNN), Support Vector Machine (SVM), Decision Trees (DT), Random Forest (RF), Logistic Regression (LR) are considered to achieve the best results. Some boosting algorithms like Extreme Gradient Boosting (XGBoost) and CatBoost are also used to improve the prediction using Artificial Neural Networks (ANN). This research also focuses on data visualization to identify patterns, trends, and outliers in a massive data set. Python and Scikit-learns are used for ML. Tensor Flow and Keras, along with Python, are used for ANN model training. The DT and RF algorithms achieved the highest accuracy of 95% among the classifiers. Meanwhile, KNN obtained a second height accuracy of 93.33%. XGBoost had a gratified accuracy of 91.67%, SVM, CATBoost, and ANN had an accuracy of 90%, and LR had 88.33% accuracy.

Keywords


Cite This Article

APA Style
Rahman, P., Rifat, A., Chy, M.I., Khan, M.M., Masud, M. et al. (2023). Machine learning and artificial neural network for predicting heart failure risk. Computer Systems Science and Engineering, 44(1), 757-775. https://doi.org/10.32604/csse.2023.021469
Vancouver Style
Rahman P, Rifat A, Chy MI, Khan MM, Masud M, Aljahdali S. Machine learning and artificial neural network for predicting heart failure risk. Comput Syst Sci Eng. 2023;44(1):757-775 https://doi.org/10.32604/csse.2023.021469
IEEE Style
P. Rahman, A. Rifat, M.I. Chy, M.M. Khan, M. Masud, and S. Aljahdali "Machine Learning and Artificial Neural Network for Predicting Heart Failure Risk," Comput. Syst. Sci. Eng., vol. 44, no. 1, pp. 757-775. 2023. https://doi.org/10.32604/csse.2023.021469



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.
  • 1894

    View

  • 903

    Download

  • 0

    Like

Share Link