Prediction of childbearing tendency in women on the verge of marriage using machine learning techniques

The declining fertility rate and increasing marriage age among girls pose challenges for policymakers, leading to issues such as population decline, higher social and economic costs, and reduced labor productivity. Using machine learning (ML) techniques to predict the desire to have children can offer a promising solution to address these challenges. Therefore, this study aimed to predict the childbearing tendency in women on the verge of marriage using ML techniques. Data from 252 participants (203 expressing a "desire to have children" and 49 indicating "reluctance to have children") in Abadan, and Khorramshahr cities (Khuzestan Province, Iran) was analyzed. Seven ML algorithms, including multilayer perceptron (MLP), support vector machine (SVM), logistic regression (LR), random forest (RF), J48 decision tree, Naive Bayes (NB), and K-nearest neighbors (KNN), were employed. The performance of these algorithms was assessed using metrics derived from the confusion matrix. The RF algorithm showed superior performance, with the highest sensitivity (99.5%), specificity (95.6%), and receiver operating characteristic curve (90.1%) values. Meanwhile, MLP emerged as the top-performing algorithm, showcasing the best overall performance in accuracy (77.75%) and precision (81.8%) compared to other algorithms. Factors such as age of marriage, place of residence, and strength of the family center with the birth of a child were the most effective predictors of a woman's desire to have children. Conversely, the number of daughters, the wife's ethnicity, and the spouse's ownership of assets such as cars and houses were among the least important factors in predicting this desire. ML algorithms exhibit excellent predictive capabilities for childbearing tendencies in women on the verge of marriage, highlighting their remarkable effectiveness. This capacity to offer accurate prognoses holds significant promise for advancing research in this field.