Comparative analysis of various supervised machine learning algorithms for the early prediction of type-II diabetes mellitus
Shahid Mohammad Ganie, Majid Bashir Malik
Abstract
Diabetes is one among the top 10 causes of death. Diabetes mellitus is a fatal disease that poses a unique and significant threat to millions of people over the globe. Despite the absolute truth about the statistical data of diabetes from various sources like the World Health Organization, International Diabetes Federation, American Diabetes Association, etc. there is a positive message that early prediction along with appropriate care, diabetes mellitus can be managed and its complications can also be prevented. Nowadays in healthcare sector, machine learning techniques are gaining immense importance through their analytical classification capabilities. Machine learning paradigms are being exploited by researchers for better prediction of diabetes to save human lives. In this paper, a comparison of different supervised machine learning classifiers based on the performance evaluation of various metrics for the early prediction of type-II diabetes mellitus (T2DM) has been performed. The experimental work has been successfully carried out using six machine learning classification algorithms. Among all classifiers, random forest (RF) performs better for predicting T2DM with an accuracy rate of 93.75%. In addition, ten-fold cross-validation method has been applied to remove the class biasness in the dataset.