Return
Data-Driven Diabetes Risk Factor Prediction Using Machine Learning Algorithms with Feature Selection Technique
DOI:10.3390/su15064930.png)
Abstract
En 中文
As type 2 diabetes becomes more prevalent across the globe, predicting its sources becomes more important. However, there is a big void in predicting the risk factors of this disease. Thus, the purpose of this study is to predict diabetes risk factors by applying machine learning (ML) algorithms. Two-fold feature selection techniques (i.e., principal component analysis, PCA, and information gain, IG) have been applied to boost the prediction accuracy. Then, the optimal features are fed into five ML algorithms, namely decision tree, random forest, support vector machine, logistic regression, and KNN. The primary data used to train the ML model were collected based on the safety procedure described in the Helsinki Declaration, 2013, and 738 records were included in the final analysis. The result has shown an accuracy level of over 82.2%, with an AUC (area under the ROC curve) value of 87.2%. This research not only identified the most important clinical and nonclinical factors in diabetes prediction, but it also found that the clinical risk factor (glucose) is the most relevant for diabetes prediction, followed by dietary factors. The noteworthy contribution of this research is the identification of previously unclassified factors left over from the previous study that considered both clinical and non-clinical aspects.
Keywords:
diabetes
feature selection
risk factors
machine learning
Journal
IF:
3.3
Papers:
10.7W
Citations:
28.4W
Organization
Cited Papers
An Ensemble Approach to Predict Early-Stage Diabetes Risk Using Machine Learning: An Empirical Study
SENSORS
IF3.5
Relationship between algae and environment: an Early Cretaceous case study, Trascău Mountains, Romania
Facies
IF0

