Machine Learning Algorithms for understanding the determinants of under-five Mortality.
Machine Learning Algorithms for understanding the determinants of under-five Mortality.
复制标题
DOI:
10.1186/s13040-022-00308-8
复制
发表时间:
2022-09-24
期刊:
影响因子:
4.5
通讯作者:
Chilyabanyama, Obvious N.
中科院分区:
文献类型:
--
作者:
Saroj, Rakesh Kumar;Yadav, Pawan Kumar;Singh, Rajneesh;Chilyabanyama, Obvious N.
Under-five mortality is a matter of serious concern for child health as well as the social development of any country. The paper aimed to find the accuracy of machine learning models in predicting under-five mortality and identify the most significant factors associated with under-five mortality. The data was taken from the National Family Health Survey (NFHS-IV) of Uttar Pradesh. First, we used multivariate logistic regression due to its capability for predicting the important factors, then we used machine learning techniques such as decision tree, random forest, Naïve Bayes, K- nearest neighbor (KNN), logistic regression, support vector machine (SVM), neural network, and ridge classifier. Each model’s accuracy was checked by a confusion matrix, accuracy, precision, recall, F1 score, Cohen’s Kappa, and area under the receiver operating characteristics curve (AUROC). Information gain rank was used to find the important factors for under-five mortality. Data analysis was performed using, STATA-16.0, Python 3.3, and IBM SPSS Statistics for Windows, Version 27.0 software. By applying the machine learning models, results showed that the neural network model was the best predictive model for under-five mortality when compared with other predictive models, with model accuracy of (95.29% to 95.96%), recall (71.51% to 81.03%), precision (36.64% to 51.83%), F1 score (50.46% to 62.68%), Cohen’s Kappa value (0.48 to 0.60), AUROC range (93.51% to 96.22%) and precision-recall curve range (99.52% to 99.73%). The neural network was the most efficient model, but logistic regression also shows well for predicting under-five mortality with accuracy (94% to 95%)., AUROC range (93.4% to 94.8%), and precision-recall curve (99.5% to 99.6%). The number of living children, survival time, wealth index, child size at birth, birth in the last five years, the total number of children ever born, mother’s education level, and birth order were identified as important factors influencing under-five mortality. The neural network model was a better predictive model compared to other machine learning models in predicting under-five mortality, but logistic regression analysis also shows good results. These models may be helpful for the analysis of high-dimensional data for health research.
登录
查看更多内容
影响因子:
3.5
作者:
Fenta HM;Zewotir T;Muluneh EK
通讯作者:
Muluneh EK
DOI:
10.1017/asb.2017.45
发表时间:
2018-05-01
期刊:
ASTIN BULLETIN
影响因子:
--
作者:
Hainaut, Donatien
通讯作者:
Hainaut, Donatien
影响因子:
2.4
作者:
Amoroso, Cheryl L.;Nisingizwe, Marie Paul;Ngabo, Fidele
通讯作者:
Ngabo, Fidele
影响因子:
2.9
作者:
Adegbosin, Adeyinka Emmanuel;Stantic, Bela;Sun, Jing
通讯作者:
Sun, Jing
影响因子:
4.8
作者:
Burges, CJC
通讯作者:
Burges, CJC