Table 2 Performance comparison of machine learning models.
Model | C | Brier | AUC | |||||||
---|---|---|---|---|---|---|---|---|---|---|
3-year | P | 5-year | P-value | 10-year | P-value | 3-year | 5-year | 10-year | ||
COX | 0.682 | 0.201 | 0.006 | 0.222 | 0.004 | 0.217 | 0.002 | 0.671 | 0.680 | 0.705 |
DT | 0.69 | 0.318 | < 0.001 | 0.349 | < 0.001 | 0.303 | < 0.001 | 0.659 | 0.659 | 0.700 |
RF | 0.701 | 0.201 | 0.007 | 0.224 | 0.008 | 0.215 | < 0.001 | 0.732 | 0.751 | 0.748 |
GBM | 0.681 | 0.208 | 0.100 | 0.230 | 0.059 | 0.239 | 0.607 | 0 0.667 | 0.680 | 0.711 |
SVM | 0.567 | 0.222 | 0.063 | 0.251 | 0.223 | 0.262 | 0.002 | 0.544 | 0.527 | 0.527 |
XGboost | 0.682 | 0.210 | 0.374 | 0.233 | 0.325 | 0.220 | 0.015 | 0.671 | 0.680 | 0.706 |