Table 6 Performance of the models—with stacking.

From: Ensemble machine learning framework for predicting maternal health risk during pregnancy

Base learners

Stacking model learner

Class

Count

TP

TN

FN

FP

 

Precision

Recall

F1

GBT

RF

DT

KNN

GBT

HR

272

246

714

28

26

 

0.897810219

0.904411765

0.901098901

LR

406

340

565

43

66

0.88772846

0.837438424

0.861850444

MR

336

282

603

75

54

0.789915966

0.839285714

0.813852814

Total

1014

868

1882

146

146

Overall → 

0.856015779

0.856015779

0.856015779

      

Weighted → 

0.858021596

0.85601578

0.856474089

GBT

RF

DT

KNN

RF

HR

272

245

714

28

27

 

0.897435897

0.900735294

0.899082569

LR

406

334

522

86

72

0.795238095

0.822660099

0.808716707

MR

336

244

601

77

92

0.760124611

0.726190476

0.742770167

Total

1014

823

1837

191

191

Overall → 

0.811637081

0.811637081

0.811637081

      

Weighted → 

0.811016864

0.81163708

0.811104752

GBT

RF

DT

KNN

DT

HR

272

242

701

41

30

 

0.855123675

0.889705882

0.872072072

LR

406

323

511

97

83

0.769047619

0.795566502

0.782082324

MR

336

228

595

83

108

0.733118971

0.678571429

0.704791345

Total

1014

793

1807

221

221

Overall → 

0.782051282

0.782051282

0.782051282

      

Weighted → 

0.780231703

0.78205128

0.780610374

GBT

RF

DT

KNN

KNN

HR

272

203

703

39

69

 

0.838842975

0.746323529

0.789883268

LR

406

295

479

129

111

0.695754717

0.726600985

0.710843373

MR

336

216

546

132

120

0.620689655

0.642857143

0.631578947

Total

1014

714

1728

300

300

Overall → 

0.704142012

0.704142012

0.704142012

      

Weighted → 

0.709263736

0.70414201

0.705780261