Title
Machine Learning and Artificial Neural Network for Predicting Heart Failure Risk
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.
Year
DOI
Venue
2023
10.32604/csse.2023.021469
COMPUTER SYSTEMS SCIENCE AND ENGINEERING
Keywords
DocType
Volume
Heart failure prediction, data visualization, machine learning, k-nearest neighbors, support vector machine, decision tree, random forest, logistic regression, xgboost and catboost, artificial neural network
Journal
44
Issue
ISSN
Citations 
1
0267-6192
0
PageRank 
References 
Authors
0.34
0
6
Name
Order
Citations
PageRank
Polin Rahman100.34
Ahmed Rifat200.34
Md IftehadAmjad Chy300.34
Mohammad Monirujjaman Khan4010.82
Mehedi Masud57726.95
Sultan Aljahdali600.34