Exploring feature selection and classification methods for predicting heart disease
TL
TL;DR
The benefits of using feature selection vary depending on the machine learning technique used for the heart datasets, and the best model created used a combination of Chi-squared feature selection with the BayesNet algorithm and achieved an accuracy of 85.00% on the considered datasets.
Abstract
Machine learning has been used successfully to improve the accuracy of computer-aided diagnosis systems. This paper experimentally assesses the performance of models derived by machine learning techniques by using relevant features chosen by various feature-selection methods. Four commonly used heart disease datasets have been evaluated using principal component analysis, Chi squared testing, ReliefF and symmetrical uncertainty to create distinctive feature sets. Then, a variety of classification algorithms have been used to create models that are then compared to seek the optimal features combinations, to improve the correct prediction of heart conditions. We found the benefits of using feature selection vary depending on the machine learning technique used for the heart datasets we consider. However, the best model we created used a combination of Chi-squared feature selection with the BayesNet algorithm and achieved an accuracy of 85.00% on the considered datasets.
References40 items
1
Digital Health
2
Prediction of Coronary Heart Disease using Machine Learning: An Experimental Analysis
3
Machine learning in autistic spectrum disorder behavioral research: A review and ways forward
4
An optimized feature selection based on genetic approach and support vector machine for heart disease
5