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.
-
Notifications
You must be signed in to change notification settings - Fork 0
dgambone3/Fetal_Health_Prediction
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Machine learning project to predict fetal health from cardiotocography results
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published