Skip to main content

Table 7 Confusion matrix of nine algorithms for 5-year BCSS and 5-year OS

From: The impact of chemotherapy and survival prediction by machine learning in early Elderly Triple Negative Breast Cancer (eTNBC): a population based study from the SEER database

Algorithms

Predictions

Algorithms

Predictions

Dead

Alive

Dead

Alive

5-year BCSS

5-year OS

K-nearest neighbor

Dead

3

47

K-nearest neighbor

Dead

15

56

Alive

7

350

Alive

17

337

Catboost

Dead

8

42

Catboost

Dead

15

56

Alive

9

348

Alive

8

346

Decision tree

Dead

13

37

Decision tree

Dead

21

50

Alive

18

339

Alive

21

333

Random forest

Dead

7

43

Random forest

Dead

18

53

Alive

10

347

Alive

16

338

Gradient booster

Dead

5

45

Gradient booster

Dead

12

59

Alive

3

354

Alive

5

349

LightGBM

Dead

5

45

LightGBM

Dead

14

57

Alive

3

354

Alive

6

348

Neural network model

Dead

0

50

Neural network model

Dead

24

47

Alive

0

357

Alive

18

336

Support vector machine

Dead

5

45

Support vector machine

Dead

11

60

Alive

3

354

Alive

2

352

XGBoost

Dead

8

42

XGBoost

Dead

18

53

Alive

7

350

Alive

4

350