Handling imbalanced class problem of measles infectionrisk prediction model

2Citations
Citations of this article
10Readers
Mendeley users who have this article in their library.
Get full text

Abstract

Measles is an emerging infectious disease with increasing number of reported cases. It is a vaccine-preventable disease;thus, it is common to have imbalanced class problem in the dataset. This study aims to resolve the imbalanced class problem for the prediction of measles infection risk and to compare the predictive results on a balanced dataset based on three machine learningtechniques. The data that was utilized in this study contained 37,884 records of suspected measles casesthat were highly imbalanced towards negative measles cases. The Synthetic Minority Over-Sampling Technique (SMOTE) was performed to balance thedistribution of the target attribute. The balanced dataset was then modelled using logistic regression, decision tree and Naïve Bayes. The predicted results indicated that logistic regression executed on the balanced dataset by SMOTE has the highest and most accurateclassification with 94.5% overall accuracy, 93.9% true positive rate, 5.8% false positive rate and 5.1% false negative rate. Therefore, SMOTE and other over-sampling approaches may be applicable to overcome imbalanced class issues in the medical dataset.

Cite

CITATION STYLE

APA

Wan Ahmad, W. M. T., Ghani, N., & Drus, S. M. (2019). Handling imbalanced class problem of measles infectionrisk prediction model. International Journal of Engineering and Advanced Technology, 9(1), 3431–3435. https://doi.org/10.35940/ijeat.A2649.109119

Register to see more suggestions

Mendeley helps you to discover research relevant for your work.

Already have an account?

Save time finding and organizing research with Mendeley

Sign up for free