Table 7 Area under curve of the predictive models based on eight machine learning algorithms for overall survival.
6-month | 12-month | 24-month | |||||||
|---|---|---|---|---|---|---|---|---|---|
Training set | Validation set | Test set | Training set | Validation set | Test set | Training set | Validation set | Test set | |
GBM | 0.837 | 0.838 | 0.856 | 0.809 | 0.735 | 0.777 | 0.750 | 0.664 | 0.691 |
RSF | 0.805 | 0.756 | 0.848 | 0.788 | 0.699 | 0.809 | 0.780 | 0.659 | 0.709 |
Lasso Cox | 0.797 | 0.814 | 0.795 | 0.762 | 0.714 | 0.737 | 0.703 | 0.659 | 0.668 |
Coxboost | 0.804 | 0.817 | 0.817 | 0.764 | 0.720 | 0.751 | 0.705 | 0.655 | 0.689 |
Survivalsvm | 0.729 | 0.740 | 0.759 | 0.707 | 0.656 | 0.716 | 0.641 | 0.623 | 0.543 |
Xgboost | 0.823 | 0.835 | 0.817 | 0.793 | 0.742 | 0.753 | 0.734 | 0.667 | 0.691 |
Superpc | 0.709 | 0.722 | 0.729 | 0.703 | 0.644 | 0.710 | 0.642 | 0.616 | 0.554 |
PlsRcox | 0.803 | 0.817 | 0.828 | 0.765 | 0.719 | 0.762 | 0.706 | 0.652 | 0.696 |