The Applicability of Some Machine Learning Algorithms in the Prediction of Type 2 Diabetes

• Published in: Proceedings of the ... International Conference on Business Excellence Vol. 18; no. 1; pp. 246 - 257
• Main Authors: Vîrgolici, Oana, Tănăsescu, Laura Gabriela
• Format: Journal Article
• Language: English
• Published: Sciendo 01.06.2024
• Subjects: Decision Tree (DT); k Nearest Neighbors (kNN); Logistic Regression (LR); Machine Learning; Support Vector Machine (SVM); type 2 diabetes mellitus
• ISSN: 2558-9652; 2502-0226; 2558-9652
• DOI: 10.2478/picbe-2024-0021

Type 2 diabetes is a metabolic disease that causes abnormal high levels of glucose in the blood. The pancreas is healthy, but the body doesn’t respond properly to its own insulin. The principal culprit is obesity, too much high fat tissue. So, measuring the body mass index or the waist circumference is a step to estimate the risk for this disease. Many people have no symptoms and the disease develops silently, causing serious problems with eyes, feet, heart and nerves. The prediction of diabetes is a very topical problem. In addition to medical guides, more and more machine learning models appear, trained on different databases. The purpose of these models is to predict diabetes, based on different parameters, not all of them coming from medical analyses. In the paper we present four diabetes prediction models, respectively based on the decision tree, support vector machine, logistic regression and k-nearest neighbors’ algorithms. All models are trained and tested on a database with approximately 65,000 records (divided into 70% for training and 30% for testing), which contains two blood markers (haemoglobin A1c and glucose), an anthropometric parameter (body mass index), age, gender and three categorical parameters (smoking status, hypertension, heart disease). We identify that Haemoglobin A1C and glucose are the most influential predictors. The models are evaluated in terms of accuracy score and confusion matrix and a ranking is presented at the end. The results obtained are very encouraging for all the presented models.