A Comparative Study of Machine Learning Algorithms for Predicting Weight Range of Neonate
A Comparative Study of Machine Learning Algorithms for Predicting Weight Range of Neonate
S. Adeeba,Banujan Kuhaneswaran,B. Kumara,S. Prasanth
Abstract
Birth weight is a crucial measure of pregnancy outcome, and it indicates a neonate’s chances of longevity, growth, long-lived health, and mental development. In epidemiological studies, it is commonly regarded as the causative pathway to these health consequences. Scanning the fetus’s weight before birth is associated with difficulties and is typically calculated using empirical calculations based on doctors' clinical experience. In this research, the authors proposed an approach to predict the neonate weight class with minimum errors and costs during six months of pregnancy, utilising existing pregnant women’s data. Authors recognised 16 attributes after an extreme literature survey and expert opinions. The haemoglobin level of the mother, mother’s pre-pregnancy weight and height, pregnant woman’s weight each month until 6 are some attributes identified. For the Machine learning (ML) models to work, a dataset of 1000 pregnant women was obtained and preprocessed. Then dataset was prepared to apply ML models. This research uses eight different ML algorithms to predict the neonate weight class at six months of pregnancy. Also, this research provides an approach to compare different eight ML models to find an effective algorithm for the neonate weight prediction. The Linear Discriminant Analysis (LDA) model has higher performance than other models. Accuracy, precision, recall and F1-score for LDA are 81.94 %, 83.05%, 94.23% and 88.28 % respectively.
