Exploring feature selection and classification methods for predicting heart disease. (March 2020)

• Record Type: Journal Article
• Authors: Spencer, Robinson; Thabtah, Fadi; Abdelhamid, Neda; Thompson, Michael
• Is Part Of: Digital health. Volume 6(2020)
• Publication Date: 2020-03
• Subjects: Classification -- data analysis -- feature selection -- heart disease -- machine learning -- prediction; Medical care -- Data processing -- Periodicals; Medical informatics -- Periodicals; 362.10285
• DOI: 10.1177/2055207620914777
• Languages: English
• ISSNs: 2055-2076

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.