Abstract

Info

Title:: Efficacy of machine learning algorithms for heart disease prediction

Author(s):: JIANG Meiyan1; 2; ZHANG Hui1; 2; 1. The First Clinical Medical College, Guangdong Medical University, Zhanjiang 524023, China 2. Department of Anesthesiology, Guangdong Second Provincial General Hospital, Guangdong Medical University, Guangzhou 510317, China

Abstract:: Objective To explore the prediction of heart diseases using machine learning-based methods, including decision trees (DT), random forest (RF), support vector machine (SVM), K-nearest neighbors (KNN), and naive Bayes (NB). Methods The Cleveland heart disease dataset was utilized as the data source. Significant features were selected using Pearson correlation coefficients. Heart disease prediction models were constructed using DT, RF, SVM, KNN, and NB algorithms, separately, and the model performance was evaluated with multiple metrics, including accuracy, precision, recall rate, F1 score, and AUC value. Results The study included 303 samples, and among the 13 clinical features, 11 were found to be significant. RF prediction model achieved the highest accuracy (0.869), recall rate (0.906), F1 score (0.879), and AUC value (0.93), while NB prediction model obtained the highest precision (0.900). Conclusion Machine learning-based methods are promising in heart disease prediction, with the RF prediction model demonstrating significant advantages and NB prediction model exhibiting satisfactory performance.