Predicting preterm birth using auto-ML frameworks: a large observational study using electronic inpatient discharge data.

ABSTRACT:

Background

To develop and compare different AutoML frameworks and machine learning models to predict premature birth.

Methods

The study used a large electronic medical record database to include 715,962 participants who had the principal diagnosis code of childbirth. Three Automatic Machine Learning (AutoML) were used to construct machine learning models including tree-based models, ensembled models, and deep neural networks on the training sample (N = 536,971). The area under the curve (AUC) and training times were used to assess the performance of the prediction models, and feature importance was computed via permutation-shuffling.

Results

The H2O AutoML framework had the highest median AUC of 0.846, followed by AutoGluon (median AUC: 0.840) and Auto-sklearn

SUBMITTER: Kong D 

PROVIDER: S-EPMC10867966 | biostudies-literature | 2024

REPOSITORIES: biostudies-literature