Performance Evaluation of Machine Learning Techniques (MLT) for Heart Disease Prediction

Abstract

The leading cause of death worldwide today is heart disease (HD). The heart is recognised as the second-most significant organ behind the brain. A successful outcome of treatment can be improved by an early diagnosis which can significantly reduce the chance of death in health care. In this paper, we proposed a method to predict heart disease. We used various machine learning algorithms (MLA), namely, logistic regression (LR), k-nearest neighbor (KNN), support vector machine (SVM), Naive Bayes (NB), random forest (RF), and decision tree (DT). With the testing data set, we evaluated the model’s accuracy in heart disease prediction. When compared to the other five models, the random forest and k-nearest neighbor approaches perform better. With a 99.04% accuracy rate, the k-nearest neighbor algorithm and random forest provide the best match to the data as compared to other algorithms. Six feature selection algorithms were used for the performance evaluation matrix. MCC parameters for accuracy, precision, recall, and F measure are used to evaluate models.

Diabetes risk prediction model based on community follow-up data using machine learning

“…Experiments showed that random forest had the highest accuracy rate of 98%, followed by multi-layer perceptron (90.99 %), Support Vector Machines (92%), Gradient Boosting (97%), Decision Trees (96%), and Logistic Regression (69%). This is similar to the findings of heart disease risk prediction studies ( Ansari et al, 2023 ). Shahadat et al ( Uddin et al, 2022 ) considered 31 disease-related variables and conducted an analysis based on the guidelines of the Elixhauser Comorbidity Index.…”

Stacking with Recursive Feature Elimination-Isolation Forest for classification of diabetes mellitus

“…One of the essential keys that has a direct impact on classification performances is the feature selection technique [ 39 ]. Feature selection eliminates unnecessary data by removing the less relevant features [ 40 ]. Many feature selection types, such as ReliefF, mutual information, and embedded methods like Lasso and Ridge, work well to reduce the number of features during classification.…”