The application of machine learning approaches to classify and predict fertility rate in Ethiopia

Abstract

Integrating machine learning (ML) models into healthcare systems is a rapidly evolving field with the potential to revolutionize care delivery. This study aimed to classify fertility rates and identify significant predictors using ML models among reproductive women in Ethiopia. This study utilized eight ML models in 5864 reproductive-age women using Ethiopian Demographic Health Survey (EDHS), 2019 data. Phyton programming language was used to develop these models. Predictors of fertility rate were determined using the feature important techniques. The performance of models was evaluated using accuracy, area under the curve (AUC), precision, recall, F1-score, specificity, and sensitivity. The mean age of participants was 32.7 (± 5.6) years. The random forest classifier (accuracy = 0.901 and AUC = 0.961) followed by a one-dimensional convolutional neural network (accuracy = 0.899 and AUC = 0.958), logistic regression (accuracy = 0.874 and AUC = 0.937), and gradient boost classifier (accuracy = 0.851 and AUC 0.927) were the top performing ML models. Family size, age, occupation, and education with an average importance score of 0.198, 0.151, 0.118, and 0.081, respectively were the top significant predictors of the fertility rate. The best ML models to classify and predict fertility rates were random forest, one-dimensional convolutional neural network, logistic regression, and gradient boost classifier. The findings on important factors of fertility rate can inform targeted public health, programs that address disparities related to family size, occupation, education, and other socioeconomic factors.

Keywords: Classification; EDHS data; Ethiopia; Fertility rate; Machine learning; Prediction.

Fig. 1

General methodology description of the proposed study.

Fig. 2

The contribution weight of data features using the SHAP bar plot (a impact of each feature and b overall weight of each feature).

Fig. 3

The ROC curve of the developed models before applying the data balancing technique.

Fig. 4

After applying the data balancing technique, the ROC curve of the developed models.

Fig. 5

Calibration curves of the developed models before (the left one) and after applying SMOTE (the right one).