Open AccessEffective predictive modelling for coronary artery diseases using support vector machine
Author(s) -
Kuncahyo Setyo Nugroho,
Anantha Yullian Sukmadewa,
Angga Vidianto,
Wayan Firdaus Mahmudy
Publication year - 2022
Publication title -
iaes international journal of artificial intelligence
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.341
H-Index - 7
eISSN - 2252-8938
pISSN - 2089-4872
DOI - 10.11591/ijai.v11.i1.pp345-355
Subject(s) - support vector machine , computer science , hyperparameter optimization , feature selection , cad , hyperparameter , resampling , artificial intelligence , machine learning , classifier (uml) , cross validation , coronary artery disease , pattern recognition (psychology) , data mining , medicine , psychiatry , engineering drawing , engineering
Coronary artery disease (CAD) is a category of cardiovascular disease that causes the highest mortality rate in the world. CAD occurs due to plaque build-up on the walls of the arteries that supply blood to the heart and other organs of the body. To control the mortality rate, a practical model that is capable of predicting CAD is needed. Machine learning approaches have been used in solving various problems in various domains, including biomedicine. However, real-world data often has an unbalanced class distribution that can interfere with classifier performance. In addition, data has many features to process. This study focuses on effective modeling capable of predicting CAD using feature selection to handle high dimensional data and feature resampling to handle unbalanced data. Feature selection is very effective by eliminating irrelevant features from the training data. Hyperparameter tuning is also done to find the best combination of parameters in support vector machines (SVM). Our results show that the SVM cross-validated ten times has a more accurate training result. Furthermore, the grid search on SVM cross-validated ten times had more accurate training model results and achieved 88% accuracy on the test data.