Skip to main content

Table 3 Model training on 68% of the dataset

From: Comparable performance of machine learning algorithms in predicting readmission and complications following total joint arthroplasty with external validation

  

Accuracy

Recall

Precision

AUROC

Outcome

Model

M

SD

LB

UB

M

SD

LB

UB

M

SD

LB

UB

M

SD

LB

UB

90-D Composite Complications

AB

0.86

0.001

0.86

0.86

0.61

0.001

0.61

0.61

0.08

0.000

0.08

0.08

0.74

0.000

0.74

0.74

Xg Boost

0.74

0.002

0.74

0.75

0.77

0.003

0.77

0.78

0.003

0.000

0.003

0.003

0.76

0.001

0.76

0.76

LR

0.81

0.01

0.79

0.83

0.38

0.004

0.37

0.39

0.07

0.002

0.06

0.07

0.60

0.002

0.60

0.61

RF

0.80

0.002

0.80

0.81

0.41

0.001

0.41

0.42

0.11

0.000

0.11

0.11

0.62

0.000

0.62

0.62

SVM

0.63

0.01

0.63

0.63

0.61

0.01

0.61

0.61

0.03

0.001

0.03

0.03

0.62

0.00

0.62

0.62

1 Layer NN

0.88

0.21

0.87

0.89

0.23

0.28

0.21

0.25

0.05

0.10

0.04

0.05

0.56

0.07

0.56

0.57

5 Layer NN

0.88

0.21

0.87

0.90

0.23

0.28

0.21

0.24

0.04

0.10

0.04

0.05

0.56

0.07

0.56

0.57

30-D Mortality

AB

0.80

0.001

0.80

0.80

0.99

0.002

0.98

0.99

0.07

0.001

0.07

0.07

0.87

0.001

0.87

0.88

Xg Boost

0.74

0.001

0.74

0.75

0.93

0.003

0.92

0.93

0.004

0.000

0.004

0.004

0.84

0.002

0.83

0.84

LR

0.62

0.01

0.61

0.63

0.941

0.02

0.89

0.99

0.07

0.002

0.06

0.07

0.77

0.01

0.75

0.80

RF

0.65

0.001

0.65

0.65

1.000

0.00

1.00

1.00

0.12

0.000

0.12

0.13

0.81

0.001

0.81

0.81

SVM

0.66

0.10

0.66

0.67

0.72

0.08

0.72

0.73

0.00

0.001

0.00

0.00

0.69

0.06

0.69

0.70

1 Layer NN

0.89

0.22

0.88

0.91

0.34

0.37

0.31

0.36

0.00

0.01

0.00

0.00

0.61

0.13

0.60

0.62

5 Layer NN

0.89

0.22

0.88

0.91

0.33

0.36

0.30

0.35

0.00

0.01

0.00

0.00

0.61

0.13

0.60

0.62

30-D Readmission

AB

0.88

0.001

0.88

0.88

0.59

0.001

0.59

0.59

0.09

0.000

0.09

0.09

0.74

0.000

0.74

0.74

GB

0.95

0.000

0.95

0.95

0.53

0.000

0.53

0.54

0.01

0.000

0.01

0.01

0.74

0.000

0.74

0.74

LR

0.85

0.001

0.85

0.85

0.34

0.001

0.34

0.34

0.07

0.000

0.07

0.07

0.60

0.000

0.60

0.60

RF

0.82

0.001

0.81

0.82

0.39

0.001

0.38

0.39

0.11

0.000

0.11

0.12

0.61

0.000

0.61

0.61

SVM

0.61

0.02

0.60

0.63

0.60

0.01

0.59

0.60

0.05

0.003

0.05

0.05

0.61

0.01

0.60

0.61

1 Layer NN

0.87

0.20

0.86

0.88

0.16

0.24

0.15

0.18

0.03

0.03

0.03

0.03

0.53

0.03

0.53

0.53

5 Layer NN

0.87

0.20

0.86

0.89

0.16

0.24

0.15

0.18

0.03

0.04

0.03

0.03

0.53

0.03

0.53

0.53

90-D Readmission

AB

0.88

0.001

0.88

0.88

0.99

0.000

0.99

0.99

0.13

0.001

0.13

0.14

0.93

0.000

0.93

0.94

GB

0.77

0.001

0.77

0.78

0.95

0.002

0.94

0.95

0.004

0.000

0.004

0.004

0.86

0.001

0.86

0.86

LR

0.80

0.01

0.78

0.83

0.99

0.000

0.99

0.99

0.14

0.01

0.13

0.15

0.89

0.01

0.88

0.91

RF

0.79

0.01

0.76

0.81

0.99

0.000

0.99

0.99

0.20

0.01

0.18

0.21

0.88

0.01

0.87

0.90

SVM

0.61

0.04

0.59

0.62

0.62

0.03

0.61

0.63

0.08

0.01

0.08

0.08

0.61

0.02

0.60

0.62

1 Layer NN

0.86

0.20

0.84

0.87

0.16

0.24

0.15

0.18

0.05

0.06

0.05

0.06

0.53

0.03

0.53

0.53

5 Layer NN

0.86

0.20

0.85

0.87

0.16

0.24

0.14

0.17

0.05

0.06

0.05

0.06

0.53

0.03

0.52

0.53

  1. LR Logistic regression, AB Adaptive boosting, Xg Boost gradient boosting, RF Random forest, SVM Support vector machine, NN a 1-layer neural network and a 5 layered NN