Birth weight prediction of new born baby with application of machine learning techniques on features of mother

Abstract

The degree of malnutrition is very high in India. Early detection of the possibility for a child to be affected by malnutrition can combat the situation to some extent. Birth weight prediction of new born baby is necessary as parent and doctors can prepare themselves for precautionary and curative measures for the development of physical and mental health. In this study, birth weight prediction of new born baby has been carried out using two machine learning techniques called Gaussian Naïve Bayes and Random Forest. These two models have been trained and tested on a self-created dataset containing 445 instances with eighteen numbers of features of mother. The dataset contains a label with two classes: low-weight and normal-weight. We got 86% accuracy for Gaussian Naïve Bayes and 100% accuracy for Random Forest. Both the techniques have shown significant improvement compared to existing studies.

Subject Classification:

• 68Q32

Keywords:

• Birth weight prediction
• Gaussian naïve bayes
• Random forest
• Features of mother
• Malnutrition
• Under-nutrition
• Nutritional status