Performance of Machine Learning Classifiers in Classifying Stunting among Under-Five Children in Zambia.

ABSTRACT: Stunting is a global public health issue. We sought to train and evaluate machine learning (ML) classification algorithms on the Zambia Demographic Health Survey (ZDHS) dataset to predict stunting among children under the age of five in Zambia. We applied Logistic regression (LR), Random Forest (RF), SV classification (SVC), XG Boost (XgB) and Naïve Bayes (NB) algorithms to predict the probability of stunting among children under five years of age, on the 2018 ZDHS dataset. We calibrated predicted probabilities and plotted the calibration curves to compare model performance. We computed accuracy, recall, precision and F1 for each machine learning algorithm. About 2327 (34.2%) children were stunted. Thirteen of fifty-eight features were selected for inclusion in the model using random fores

SUBMITTER: Chilyabanyama ON 

PROVIDER: S-EPMC9320245 | biostudies-literature | 2022 Jul

REPOSITORIES: biostudies-literature