English

Machine Learning Classifications of Coronary Artery Disease

Machine Learning 2018-12-10 v1

Abstract

Coronary Artery Disease (CAD) is one of the leading causes of death worldwide, and so it is very important to correctly diagnose patients with the disease. For medical diagnosis, machine learning is a useful tool, however features and algorithms must be carefully selected to get accurate classification. To this effect, three feature selection methods have been used on 13 input features from the Cleveland dataset with 297 entries, and 7 were selected. The selected features were used to train three different classifiers, which are SVM, Na\"ive Bayes and KNN using 10-fold cross-validation. The resulting models evaluated using Accuracy, Recall, Specificity and Precision. It is found that the Na\"ive Bayes classifier performs the best on this dataset and features, outperforming or matching SVM and KNN in all the four evaluation parameters used and achieving an accuracy of 84%.

Keywords

Cite

@article{arxiv.1812.02828,
  title  = {Machine Learning Classifications of Coronary Artery Disease},
  author = {Ali Bou Nassif and Omar Mahdi and Qassim Nasir and Manar Abu Talib and Mohammad Azzeh},
  journal= {arXiv preprint arXiv:1812.02828},
  year   = {2018}
}

Comments

6 pages, The 13th International Joint Symposium on Artificial Intelligence and Natural Language Processing (iSAI-NLP 2018)