This is a basic neural network with two hidden layers and tunable hyperparameters and cost functions, built without the use of keras or tensorflow (sklearn is used to split the data into train and test).
LBW_Dataset.csv contains the Low Birth weigh dataset, which is small dataset of just 96 records. LBW is closely associated with infant mortality as well as various health outcomes later in life. Various studies show strong correlation between maternal health during pregnancy and the child’s birth weight. We use health indicators of pregnant women such as age, height, weight, community etc in order for early detection of potential LBW cases. This detection is treated as a classification problem between LBW and not-LBW classes. The problem statement was to build a model to classify patient cases into LBW and non-LBW classes based on these indicators.
The Dataset consists of 10 columns
- Community - Information about the patient’s community
- SC - 1
- ST - 2
- BC - 3
- OC - 4
- Age - Patients age in years
- Weight - Weight in Kg during Trimester
- Delivery Phase -
- 1 - Before 37 weeks
- 2 - After 42 weeks
- HB - Haemoglobin content
- IFA - determines if the patient took Folic acid or not
- 1 - patient consumed Folic acid
- 0 - patient did not consume Folic acid
- BP - Blood Pressure during Trimester
- Education - Educational Qualification of the patient on a scale of 0-10
- Residence - indicates whether the patient is resident of the town or village in which the study was conducted (indicated by 1) or if the patient is a non-resident (indicated by 2)
- Result - Label 1 indicates case of LBW, Label 0 indicates non LBW case
- LBW_Dataset.csv - The dataset as mentioned above
- pre-processing.py - Cleans the data
- preprocessed.csv - Generated by pre-processing.py
- Neural_Net.py - Trains the model, performs testing and prints results