Comparison of three data mining models for predicting diabetes or prediabetes by risk factors.
The purpose of this study was to compare the performance of logistic regression, artificial neural networks (ANNs) and decision tree models for predicting diabetes or prediabetes using common risk factors. Participants came from two communities in Guangzhou, China; 735 patients confirmed to have diabetes or prediabetes and 752 normal controls were recruited. A standard questionnaire was administered to obtain information on demographic characteristics, family diabetes history, anthropometric measurements and lifestyle risk factors. Then we developed three predictive models using 12 input variables and one output variable from the questionnaire information; we evaluated the three models in terms of their accuracy, sensitivity and specificity. The logistic regression model achieved a classification accuracy of 76.13% with a sensitivity of 79.59% and a specificity of 72.74%. The ANN model reached a classification accuracy of 73.23% with a sensitivity of 82.18% and a specificity of 64.49%; and the decision tree (C5.0) achieved a classification accuracy of 77.87% with a sensitivity of 80.68% and specificity of 75.13%. The decision tree model (C5.0) had the best classification accuracy, followed by the logistic regression model, and the ANN gave the lowest accuracy.
Department of Health Service Management, Public Health School of Sun Yat-Sen University, People's Republic of China.
Xue-Hui
XH
Meng
Yi-Xiang
YX
Huang
Dong-Ping
DP
Rao
Qiu
Q
Zhang
Qing
Q
Liu
D000328:Adult; D000368:Aged; D000369:Aged, 80 and over; D057225:Data Mining / Q000706:statistics & numerical data*; D003663:Decision Trees; D003920:Diabetes Mellitus / Q000175:diagnosis*; D005260:Female; D006801:Humans; D016015:Logistic Models; D008297:Male; D008875:Middle Aged; D016571:Neural Networks, Computer; D011236:Prediabetic State / Q000175:diagnosis*; D011379:Prognosis; D012307:Risk Factors; D012680:Sensitivity and Specificity; D011795:Surveys and Questionnaires