Dengue classification method using support vector machines and cross-validation techniques

Hamdani Hamdani, Heliza Rahmania Hatta, Novianti Puspitasari, Anindita Septiarini, Henderi Henderi

Abstract


Dengue is a dangerous disease that can lead to death if the diagnosis and treatment are inappropriate. The common symptoms that occur, including headache, muscle aches, fever, and rash. Dengue is a disease that causes endemics in several countries in South Asia and Southeast Asia. There are three varieties of dengue, such as dengue fever (DF), dengue hemorrhagic fever (DHF), and dengue shock syndrome (DSS). This disease can currently be classified using a machine learning approach with the input data being the dengue symptoms. This study aims to classify dengue types consisting of three classes: DF, DHF, and DSS using five classification methods including C.45, decision tree (DT), k-nearest neighbor (KNN), random forest (RF), and support vector machine (SVM). The dataset used consists of 21 attributes, which are the dengue symptoms. It was collected from 110 patients. The evaluation method was conducted using cross-validation with k-folds of 3, 5, and 10. The dengue classification method was evaluated using three parameters: precision, recall, and accuracy, which were most optimally achieved. The most optimal evaluation results were obtained using SVM with k-fold 3 and 10 with precision, recall, and accuracy values reaching 99.1%, 99.1%, and 99.1%, respectively.

Keywords


classification; cross validation; dengue; machine learning; support vector machine;

Full Text:

PDF


DOI: http://doi.org/10.11591/ijai.v11.i3.pp1119-1129

Refbacks

  • There are currently no refbacks.


View IJAI Stats

Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.