Skip to content

Machine learning project to predict fetal health from cardiotocography results

Notifications You must be signed in to change notification settings

dgambone3/Fetal_Health_Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Predicting Fetal Health

Cardiotocography is a powerful tool that is mainly used to determine how much oxygen the fetus is receiving, which is crucial for development. Classification models decision tree, support vector machine, gradient boost, k-nearest neighbors and logistic regression were applied to a multi-class dataset predict fetal health. 10-fold cross validation was used along with a grid search to perform hyper-parameter tuning. It was determined that gradient boost was the best performing model given this dataset, with a 0.99 value for area under the ROC curve.

Preprocessing and Exploratory Data Analysis

Box-Plots

Normalized Box-Plots

Histograms

Correlation Matrix

Principal Component Analysis

Results

Decision Tree Classifier

Support Vector Machine

Gradient Boosting

k-Nearest Neighbors

Logistic Regression