NYC Data Science Academy| Blog
Bootcamps
Lifetime Job Support Available Financing Available
Bootcamps
Data Science with Machine Learning Flagship ๐Ÿ† Data Analytics Bootcamp Artificial Intelligence Bootcamp New Release ๐ŸŽ‰
Free Lesson
Intro to Data Science New Release ๐ŸŽ‰
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook Graduate Outcomes Must See ๐Ÿ”ฅ
Alumni
Success Stories Testimonials Alumni Directory Alumni Exclusive Study Program
Courses
View Bundled Courses
Financing Available
Bootcamp Prep Popular ๐Ÿ”ฅ Data Science Mastery Data Science Launchpad with Python View AI Courses Generative AI for Everyone New ๐ŸŽ‰ Generative AI for Finance New ๐ŸŽ‰ Generative AI for Marketing New ๐ŸŽ‰
Bundle Up
Learn More and Save More
Combination of data science courses.
View Data Science Courses
Beginner
Introductory Python
Intermediate
Data Science Python: Data Analysis and Visualization Popular ๐Ÿ”ฅ Data Science R: Data Analysis and Visualization
Advanced
Data Science Python: Machine Learning Popular ๐Ÿ”ฅ Data Science R: Machine Learning Designing and Implementing Production MLOps New ๐ŸŽ‰ Natural Language Processing for Production (NLP) New ๐ŸŽ‰
Find Inspiration
Get Course Recommendation Must Try ๐Ÿ’Ž An Ultimate Guide to Become a Data Scientist
For Companies
For Companies
Corporate Offerings Hiring Partners Candidate Portfolio Hire Our Graduates
Students Work
Students Work
All Posts Capstone Data Visualization Machine Learning Python Projects R Projects
Tutorials
About
About
About Us Accreditation Contact Us Join Us FAQ Webinars Subscription An Ultimate Guide to
Become a Data Scientist
    Login
NYC Data Science Acedemy
Bootcamps
Courses
Students Work
About
Bootcamps
Bootcamps
Data Science with Machine Learning Flagship
Data Analytics Bootcamp
Artificial Intelligence Bootcamp New Release ๐ŸŽ‰
Free Lessons
Intro to Data Science New Release ๐ŸŽ‰
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook
Graduate Outcomes Must See ๐Ÿ”ฅ
Alumni
Success Stories
Testimonials
Alumni Directory
Alumni Exclusive Study Program
Courses
Bundles
financing available
View All Bundles
Bootcamp Prep
Data Science Mastery
Data Science Launchpad with Python NEW!
View AI Courses
Generative AI for Everyone
Generative AI for Finance
Generative AI for Marketing
View Data Science Courses
View All Professional Development Courses
Beginner
Introductory Python
Intermediate
Python: Data Analysis and Visualization
R: Data Analysis and Visualization
Advanced
Python: Machine Learning
R: Machine Learning
Designing and Implementing Production MLOps
Natural Language Processing for Production (NLP)
For Companies
Corporate Offerings
Hiring Partners
Candidate Portfolio
Hire Our Graduates
Students Work
All Posts
Capstone
Data Visualization
Machine Learning
Python Projects
R Projects
About
Accreditation
About Us
Contact Us
Join Us
FAQ
Webinars
Subscription
An Ultimate Guide to Become a Data Scientist
Tutorials
Data Analytics
  • Learn Pandas
  • Learn NumPy
  • Learn SciPy
  • Learn Matplotlib
Machine Learning
  • Boosting
  • Random Forest
  • Linear Regression
  • Decision Tree
  • PCA
Interview by Companies
  • JPMC
  • Google
  • Facebook
Artificial Intelligence
  • Learn Generative AI
  • Learn ChatGPT-3.5
  • Learn ChatGPT-4
  • Learn Google Bard
Coding
  • Learn Python
  • Learn SQL
  • Learn MySQL
  • Learn NoSQL
  • Learn PySpark
  • Learn PyTorch
Interview Questions
  • Python Hard
  • R Easy
  • R Hard
  • SQL Easy
  • SQL Hard
  • Python Easy
Data Science Blog > Capstone > Data: Finding factors influencing NICU admissions

Data: Finding factors influencing NICU admissions

Marek Kwasnica, Mohamad Sayed, Richard Kim and Connor Haas
Posted on Apr 6, 2020
The skills I demoed here can be learned through taking Data Science with Machine Learning bootcamp with NYC Data Science Academy.

Introduction:

Every year more than 4 million families in the United States visit maternity wards bringing new children into the world. Ten percent of mothers experience complications resulting in newborns being admitted into the intensive care unit (ICU). Using annual data from the Center for Disease Control and Prevention (CDC). Our goal is to understand trends and relationships between pregnancy risk factors affecting delivery complications, congenital abnormalities, and admittance to the neonatal intensive care unit (NICU).

Using a Tableau dashboard, our analysis will show predictive models that provide insight for healthcare policy makers. This application will help systemically vulnerable populations, hospitals and primary care providers to optimize the distribution of limited resources. Through this we can identify preventative measures for patients allowing expecting families to better prepare for the challenges of pregnancy.

The Data set: Extraction and Preparation

The CDC annually publishes information for all births and pregnancy-related events for independent research and analysis. Between the years of 2014-2018 there were approximately four million births. Our team collected data covering over 200 features characterizing parents, methods of delivery, health of the newborn and admittance to the NICU or ICU for the mother which contained approximately 25GB of data.

Our data was configured into a fixed width format which required position encoding along lengths of characters. Each year had a PDF user guide detailing the file layout, encoding of features and detailed technical notes explaining data collection and imputation. 

Our application tailored parses to extract each year from the original files into pandas dataframes and saved them into CSV files for manipulation. Each dataset contained approximately 20 million observations. We downsampled each to 5%, which was 200,000 observations from each year.

This factor was decided using a second order classification using mean and standard deviation which observed no statistically significant differences between distributions within numerical and categorical features. From this raw selection we observed a marginal amount (less than 3%) of observations in boolean features were missing or unknown. Where applicable, missing or unknown data for boolean features was ignored. 

After extraction and removal of redundant features, the remaining 109 columns were analyzed in depth with respect to NICU admittance.

Data Analysis:

Between 2014 and 2018 the average rate of live birth admittance to the NICU was 8.65%. The graph below shows there is an increasing admittance to the NICU over time. While some of this increase may be due to improvements in record keeping reducing the number of cases labeled unknown, this is a worrying statistic for families due to the emotional, physical and financial cost involved.

data

โ€‹Figure 1. Yearly NICU Admittance

With this in mind, the remaining features have been divided to consider factors occurring during pregnancy, during delivery, and factors measured after delivery. Each group of features is examined first for relevance to NICU admittance, and then the factors from the first group are re-examined for ties to the most relevant factors in the latter two groups. 

Figure 2 shows lower admittance to the NICU of women giving birth aged 20-29, while women over the age of 40 have high admittance. The time series indicates that mothers in their 30s are increasing, while the number of teenage mothers is decreasing. This trend suggests that the rising rate of NICU admittances is likely to continue. Additionally observe that the number of mothers in the highest risk group, those over 40, is two orders of magnitude fewer than the lowest risk group, mothers in their 20s.

Figure 2a: NICU Admittance over time per Mother's Age

โ€‹โ€‹Figure 2b: NICU Admittance per Mother's Age

Body Mass Index Data

Following age, the next feature examined which broadly relates to overall health is the body mass index (BMI), calculated using height and weight. BMI is often used as a baseline indicator for long term health risks. Being within the โ€˜normalโ€™  BMI range has a slight protective effect against NICU admission, while any level of obesity is associated with an up to 3.4 % increase, as shown in Figure 3b. It is worth noting that being under or over the ideal range had no significant effect, and no data relating to whether dietary or nutritional needs are met is collected by the CDC.

Figure 3a: Count of Motherโ€™s BMI
Figure 3b: Percentage of Motherโ€™s BMI

In figure 4 below, observe that there does not appear to be a great risk posed by smoking before becoming pregnant. However, smoking during the first two trimesters of pregnancy slightly increases the risk of admittance to the NICU to 9.78%, and smoking during the third trimester further increases the risk to 15.61%. While getting smokers to stop smoking in general is a desirable community healthcare goal, it appears providing alternatives to cigarettes to women as they enter their third trimester is of particular importance.

โ€‹โ€‹4 : Smoking Risk by Trimester

5a : Count of Motherโ€™s Medical History Risk Factors
5b : Percentage of Motherโ€™s Medical History Risk Factors
6a : Count of Motherโ€™s Medical History Infection

โ€‹6b : Percentage of Motherโ€™s Medical History Infections

7a : Count of Infant Congenital Factors

Figure 7b : Count of Infant Congenital Factors

Pre-Existing Conditions Data

We compared over thirty pre existing conditions and they mostly remained constant each year showing little trends increasing or decreasing. One feature we saw increasing was multiple births which has been going down over time. Our goal is to find which features contributed most to an infant being admitted to the NICU. We also want to create a model that could help predict admittance. For this we decided to use two basic classification models to test performance.

Regression:

Having developed some intuition from our graphical analysis we were able to identify some features that were more predictive of NICU admittance. We wanted to validate these assumptions of feature importance and classification using a machine learning model. 

As an initial simple predictor, we trained a Logistic Regression model to predict admittance to the NICU. Since Logistic Regression can only classify between two classes, we used the one-vs-all method to create three sub-models, one for each of the possible outcomes(โ€˜Yesโ€™, โ€˜Noโ€™, โ€˜Unknownโ€™), that predicts the probability that a given baby will fall into each category(eg โ€˜Yesโ€™ vs. โ€˜Not Yesโ€™, โ€˜Noโ€™ vs. โ€˜Not Noโ€™). After calculating three probabilities with these models, the model uses outcomes with the highest probability to select the features it has found to be most important

Mother Data

However, the results of this model proved not very predictive or reproducible. In Table 1, the results are reported in a matrix showing the percentage of predictions against the actual admittance data. The sensitivity, or ability of the model to predict NICU admittance in cases where a baby was indeed admitted to the NICU, of the resulting model is about 20%. This indicated that the model heavily favored predicting the majority class โ€˜Noโ€™ regardless of the underlying data.

Repeated attempts to test the model on different test subsets of the data yielded inconsistent feature rankings. This instability in results indicated that the relationships between our dependent and independent features were likely not linear and reinforced the suspicion that the model was biased towards the majority class.

Table 1 - Confusion Matrix for Regression Model

Random Forest:

Our regression model proved to have volatile results and we wanted to try a model that was more robust. We used a random forest classifier because we found the data did not have a linear trend. We initially performed the random forest classifier on a subset of our down-sampled data frame. This was necessary due to the large size and high cost of a random forest.

Our results provided a model with 92% accuracy but after looking at the confusion matrix, we found our model was overfit and only chose the option not to admit to the NICU. We decided we wanted to create a new down-sampled data frame that had equal amounts of observations admitted to the NICU and not. After training this model we found it had 84% accuracy but our modelโ€™s sensitivity was correctly predicting 80% of cases that should be admitted to the NICU.

Table 2 - Confusion Matrix for Random Forest Model

Our top features selected by the model correspond with hospital standard practice of admission due to low birth weight, preterm and late-term birth and serious health conditions. This model proved to be more robust and fairly accurate given the complexity of features.

Conclusion:

Our analysis and modeling suggest that 10 features are likely targets for high NICU admittance. BMI and smoking while pregnant are 2 conditions that highly contribute to NICU admittance that can be addressed before and during pregnancy. Women over 40 have over twice as much risk of being having issues and women 20-30 years of age have the least risk.

The random forest model had an 80% sensitivity for predicting a child's risk based on the mothers health conditions of admittance. Our tableau application neatly displays conditions that should be considered higher risk that might contribute to the child being admitted to the NICU.

Citations:


  1. https://www.managedcaremag.com/archives/2010/1/how-plans-can-improve-outcomes-and-cut-costs-preterm-infant-care

About Authors

Marek Kwasnica

Marek is currently a data science fellow at NYC Data Science Academy. He has several years experience in biomedical engineering research. He holds a Masters of Engineering in Biological Engineering from Cornell University. Marek is passionate about applying...
View all posts by Marek Kwasnica >

Mohamad Sayed

View all posts by Mohamad Sayed >

Richard Kim

View all posts by Richard Kim >

Connor Haas

Looking for a new opportunity, I recently graduated from a data science fellowship in Manhattan. I am an electrical engineering graduate with a computer science minor. I worked as an Inside Sales Engineer in a small company that...
View all posts by Connor Haas >

Leave a Comment

No comments found.

View Posts by Categories

All Posts 2399 posts
AI 7 posts
AI Agent 2 posts
AI-based hotel recommendation 1 posts
AIForGood 1 posts
Alumni 60 posts
Animated Maps 1 posts
APIs 41 posts
Artificial Intelligence 2 posts
Artificial Intelligence 2 posts
AWS 13 posts
Banking 1 posts
Big Data 50 posts
Branch Analysis 1 posts
Capstone 206 posts
Career Education 7 posts
CLIP 1 posts
Community 72 posts
Congestion Zone 1 posts
Content Recommendation 1 posts
Cosine SImilarity 1 posts
Data Analysis 5 posts
Data Engineering 1 posts
Data Engineering 3 posts
Data Science 7 posts
Data Science News and Sharing 73 posts
Data Visualization 324 posts
Events 5 posts
Featured 37 posts
Function calling 1 posts
FutureTech 1 posts
Generative AI 5 posts
Hadoop 13 posts
Image Classification 1 posts
Innovation 2 posts
Kmeans Cluster 1 posts
LLM 6 posts
Machine Learning 364 posts
Marketing 1 posts
Meetup 144 posts
MLOPs 1 posts
Model Deployment 1 posts
Nagamas69 1 posts
NLP 1 posts
OpenAI 5 posts
OpenNYC Data 1 posts
pySpark 1 posts
Python 16 posts
Python 458 posts
Python data analysis 4 posts
Python Shiny 2 posts
R 404 posts
R Data Analysis 1 posts
R Shiny 560 posts
R Visualization 445 posts
RAG 1 posts
RoBERTa 1 posts
semantic rearch 2 posts
Spark 17 posts
SQL 1 posts
Streamlit 2 posts
Student Works 1687 posts
Tableau 12 posts
TensorFlow 3 posts
Traffic 1 posts
User Preference Modeling 1 posts
Vector database 2 posts
Web Scraping 483 posts
wukong138 1 posts

Our Recent Popular Posts

AI 4 AI: ChatGPT Unifies My Blog Posts
by Vinod Chugani
Dec 18, 2022
Meet Your Machine Learning Mentors: Kyle Gallatin
by Vivian Zhang
Nov 4, 2020
NICU Admissions and CCHD: Predicting Based on Data Analysis
by Paul Lee, Aron Berke, Bee Kim, Bettina Meier and Ira Villar
Jan 7, 2020

View Posts by Tags

#python #trainwithnycdsa 2019 2020 Revenue 3-points agriculture air quality airbnb airline alcohol Alex Baransky algorithm alumni Alumni Interview Alumni Reviews Alumni Spotlight alumni story Alumnus ames dataset ames housing dataset apartment rent API Application artist aws bank loans beautiful soup Best Bootcamp Best Data Science 2019 Best Data Science Bootcamp Best Data Science Bootcamp 2020 Best Ranked Big Data Book Launch Book-Signing bootcamp Bootcamp Alumni Bootcamp Prep boston safety Bundles cake recipe California Cancer Research capstone car price Career Career Day ChatGPT citibike classic cars classpass clustering Coding Course Demo Course Report covid 19 credit credit card crime frequency crops D3.js data data analysis Data Analyst data analytics data for tripadvisor reviews data science Data Science Academy Data Science Bootcamp Data science jobs Data Science Reviews Data Scientist Data Scientist Jobs data visualization database Deep Learning Demo Day Discount disney dplyr drug data e-commerce economy employee employee burnout employer networking environment feature engineering Finance Financial Data Science fitness studio Flask flight delay football gbm Get Hired ggplot2 googleVis H20 Hadoop hallmark holiday movie happiness healthcare frauds higgs boson Hiring hiring partner events Hiring Partners hotels housing housing data housing predictions housing price hy-vee Income industry Industry Experts Injuries Instructor Blog Instructor Interview insurance italki Job Job Placement Jobs Jon Krohn JP Morgan Chase Kaggle Kickstarter las vegas airport lasso regression Lead Data Scienctist Lead Data Scientist leaflet league linear regression Logistic Regression machine learning Maps market matplotlib Medical Research Meet the team meetup methal health miami beach movie music Napoli NBA netflix Networking neural network Neural networks New Courses NHL nlp NYC NYC Data Science nyc data science academy NYC Open Data nyc property NYCDSA NYCDSA Alumni Online Online Bootcamp Online Training Open Data painter pandas Part-time performance phoenix pollutants Portfolio Development precision measurement prediction Prework Programming public safety PwC python Python Data Analysis python machine learning python scrapy python web scraping python webscraping Python Workshop R R Data Analysis R language R Programming R Shiny r studio R Visualization R Workshop R-bloggers random forest Ranking recommendation recommendation system regression Remote remote data science bootcamp Scrapy scrapy visualization seaborn seafood type Selenium sentiment analysis sentiment classification Shiny Shiny Dashboard Spark Special Special Summer Sports statistics streaming Student Interview Student Showcase SVM Switchup Tableau teachers team team performance TensorFlow Testimonial tf-idf Top Data Science Bootcamp Top manufacturing companies Transfers tweets twitter videos visualization wallstreet wallstreetbets web scraping Weekend Course What to expect whiskey whiskeyadvocate wildfire word cloud word2vec XGBoost yelp youtube trending ZORI

NYC Data Science Academy

NYC Data Science Academy teaches data science, trains companies and their employees to better profit from data, excels at big data project consulting, and connects trained Data Scientists to our industry.

NYC Data Science Academy is licensed by New York State Education Department.

Get detailed curriculum information about our
amazing bootcamp!

Please enter a valid email address
Sign up completed. Thank you!

Offerings

  • HOME
  • DATA SCIENCE BOOTCAMP
  • ONLINE DATA SCIENCE BOOTCAMP
  • Professional Development Courses
  • CORPORATE OFFERINGS
  • HIRING PARTNERS
  • About

  • About Us
  • Alumni
  • Blog
  • FAQ
  • Contact Us
  • Refund Policy
  • Join Us
  • SOCIAL MEDIA

    ยฉ 2025 NYC Data Science Academy
    All rights reserved. | Site Map
    Privacy Policy | Terms of Service
    Bootcamp Application