Comparison of machine learning algorithms for dengue virus (DENV) classification

Abstract

Dengue, transmitted by the Aedes aegypti mosquito, necessitates accurate case classification for effective management. The research focuses on leveraging machine learning algorithms to enhance diagnosis and streamline control strategies. Utilizing a dataset from a public hospital, encompassing 21,157 cases categorized by period, outcome, gender, age, symptoms, and origin (autochthonous or imported), the study conducted a comparative analysis of Support Vector Machine, Random Forest, and Artificial Neural Network algorithms. The dataset was divided into 70% (14,809 cases) for training and 30% (6,348 cases) for testing. Results unveiled the Artificial Neural Network as the frontrunner, exhibiting an impressive 86.47% accuracy and a robust 92.91% recall in the classification of dengue-related cases. This underscores the potential of machine learning in refining dengue diagnosis and facilitating more efficient control strategies, offering a promising avenue for combating this vector-borne disease.