Prediction of heart disease and classifiers’ sensitivity analysis

• Published in: BMC bioinformatics 2020-07, Vol.21 (1), p.1-278, Article 278
• Main Author: Almustafa, Khaled Mohamad
• Format: Article
• Language: English
• Subjects: Algorithms; Bayesian analysis; Cardiovascular disease; Cardiovascular diseases; Classification; Classifiers; Comparative analysis; Coronary artery disease; Datasets; Decision tree J48; Decision trees; Feature extraction; Health care industry; Heart disease (HD); Heart diseases; High definition television; K-nearest neighbor; Machine learning; Methodology; Neural networks; Performance enhancement; Prediction; Sensitivity analysis; Stochasticity; Support vector machine (SVM); Support vector machines
• ISSN: 1471-2105; 1471-2105
• DOI: 10.1186/s12859-020-03626-y

Heart disease (HD) is one of the most common diseases nowadays, and an early diagnosis of such a disease is a crucial task for many health care providers to prevent their patients for such a disease and to save lives. In this paper, a comparative analysis of different classifiers was performed for the classification of the Heart Disease dataset in order to correctly classify and or predict HD cases with minimal attributes. The set contains 76 attributes including the class attribute, for 1025 patients collected from Cleveland, Hungary, Switzerland, and Long Beach, but in this paper, only a subset of 14 attributes are used, and each attribute has a given set value. The algorithms used K- Nearest Neighbor (K-NN), Naive Bayes, Decision tree J48, JRip, SVM, Adaboost, Stochastic Gradient Decent (SGD) and Decision Table (DT) classifiers to show the performance of the selected classifications algorithms to best classify, and or predict, the HD cases. It was shown that using different classification algorithms for the classification of the HD dataset gives very promising results in term of the classification accuracy for the K-NN (K = 1), Decision tree J48 and JRip classifiers with accuracy of classification of 99.7073, 98.0488 and 97.2683% respectively. A feature extraction method was performed using Classifier Subset Evaluator on the HD dataset, and results show enhanced performance in term of the classification accuracy for K-NN (N = 1) and Decision Table classifiers to 100 and 93.8537% respectively after using the selected features by only applying a combination of up to 4 attributes instead of 13 attributes for the predication of the HD cases. Different classifiers were used and compared to classify the HD dataset, and we concluded the benefit of having a reliable feature selection method for HD disease prediction with using minimal number of attributes instead of having to consider all available ones.