Table 5 Performance of the predictive models for classifying low-birth weight (LBW) and normal-birth weight (NBW). Individual recall for each class is presented, along with macro accuracy, recall, macro precision, macro F1-score, macro area under the receiver operating characteristic curve (ROC AUC), and macro area under the precision-recall curve (PR AUC)
From: Exploring parental factors influencing low birth weight on the 2022 CDC natality dataset
Model | LBW recall (\(\%\)) | NBW recall (\(\%\)) | Accuracy (%) | Macro recall (\(\%\)) | Macro precision (\(\%\)) | Macro F1-score (\(\%\)) | Macro ROC AUC (\(\%\)) | Macro PR AUC (\(\%\)) |
|---|---|---|---|---|---|---|---|---|
LR | 64.0 | 66.0 | 66.0 | 65.0 | 52.0 | 44.0 | 70.4 | 52.9 |
RF | 62.0 | 66.0 | 66.0 | 64.0 | 52.0 | 44.0 | 69.5 | 52.7 |
XGBoost | 66.0 | 68.0 | 68.3 | 67.0 | 52.0 | 46.0 | 73.4 | 53.9 |
CIT | 61.6 | 61.8 | 61.8 | 61.7 | 51.4 | 43.4 | 63.8 | 51.3 |
Attention Mechanism | 64.0 | 66.0 | 66.3 | 65.0 | 52.0 | 45.0 | 70.5 | 52.9 |
Average | 63.5 | 65.6 | 65.7 | 64.3 | 51.9 | 44.5 | 69.52 | 52.7 |