Diabetes Risk Prediction using Feature Importance Extreme Gradient Boosting (XGBoost)

Kartina Diah Kusuma Wardani,Memen Akbar
DOI: https://doi.org/10.29207/resti.v7i4.4651
2023-08-12
Jurnal RESTI (Rekayasa Sistem dan Teknologi Informasi)
Abstract:Diabetes results from impaired pancreas function as a producer of insulin and glucagon hormones, which regulate glucose levels in the blood. People with diabetes today are not only experienced adults, but pre-diabetes has been identified since the age of children and adolescents. Early prediction of diabetes can make it easier for doctors and patients to intervene as soon as possible so that the risk of complications can be reduced. One of the uses of medical data from diabetes patients is used to produce a model that can be used by medical staff to predict and identify diabetes in patients. Various techniques are used to provide the earliest possible prediction of diabetes based on the symptoms experienced by diabetic patients, including using machine learning. People can use Machine Learning to generate models based on historical data of diabetic patients, and predictions are made with the model. In this study, extreme gradient boosting is the machine learning technique to predict diabetes (xgboost) using Feature Importance XGBoost. The diabetes dataset used in this study comes from the Early stage diabetes risk prediction dataset published by UCI Machine Learning, which has 520 records and 16 attributes. The diabetes prediction model using xgboost is displayed as a tree. The model accuracy result in this study was 98.71%, for the F1 score was 98.18%. While the accuracy obtained based on the best 10 attributes using the XGBoost feature importance are 98.72%.
What problem does this paper attempt to address?