Skip to main content
. 2023 Jun 21;21:404. doi: 10.1186/s12967-023-04277-2

Table 4.

Performance of prognostic models built by machine learning algorithms on test data (area under the ROC curve)

6-month survival 1-year survival 2-year survival 3-year survival
XGBoost 0.824 0.813 0.800 0.803
LR 0.794 0.744 0.740 0.744
RF 0.770 0.729 0.730 0.756
SVM 0.730 0.647 0.525 0.509
KNN 0.738 0.623 0.581 0.586
ID3 0.692 0.628 0.685 0.639

XGBoost extreme gradient boosting, LR logistic regression, RF random forest, SVM support vector machine, ID3 decision tree, KNN K-Nearest Neighbor