Optimized decision tree classification method for diabetes prediction

Elly Muningsih, Fabriyan Fandi Dwi Imaniawan, Aprih Widayanto, Eva Argarini Pratama, Sutrisno Sutrisno, Sri Kiswati


Diabetes is one of the most deadly chronic diseases because most sufferers do not realize they have it. A more accurate prediction of diabetes disease must be made to reduce the risk of bad things happening to sufferers. This research will optimize the decision tree (DT) classification method for diabetes prediction. Optimization is done by splitting criteria, splitting data, particle swarm optimization (PSO), and parameter optimization to find the highest and most accurate forecast of diabetes. Splitting criteria is done by comparing the results of three criteria, namely gain ratio (GR), information gain (IG), and gini index (GI). Splitting data is done by dividing training data and testing data into three comparison groups, namely 70:30, 80:20, and 90:10. The application of PSO and parameter optimization is carried out to increase the accuracy value. The processed data is taken from the UCI machine learning repository with 520 records and 17 attributes (1 class/label attribute). From the experiments, the GI criterion with splitting data 90:10 obtained the greatest accuracy of 98.08%, and the combination with PSO resulted in an accuracy of 97.66%. Meanwhile, parameter optimization with splitting data 90:10 combined with GR criteria resulted in the highest accuracy of 97.90%.


decision tree; diabetes prediction; highest accuracy; particle swarm optimization; parameter optimization;

Full Text:


DOI: http://doi.org/10.12928/telkomnika.v22i4.25656


  • There are currently no refbacks.

Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

TELKOMNIKA Telecommunication, Computing, Electronics and Control
ISSN: 1693-6930, e-ISSN: 2302-9293
Universitas Ahmad Dahlan, 4th Campus
Jl. Ringroad Selatan, Kragilan, Tamanan, Banguntapan, Bantul, Yogyakarta, Indonesia 55191
Phone: +62 (274) 563515, 511830, 379418, 371120
Fax: +62 274 564604