Table 2.
Chemotherapy response and metastasis prediction model with various machine learning algorithms and deep learning.
Chemotherapy Response | Random Forest | Gradient Boosting | Deep Learning | ||
Text Feature (47) | AUC > 0.6 (7) | Text Feature (47) | AUC > 0.6 (7) | Train (37): Test (15) | |
Sensitivity | 0.76 | 0.79 | 0.85 | 0.84 | 0.956 |
Specificity | 0.74 | 0.82 | 0.94 | 0.88 | 0.964 |
AUC | 0.76 | 0.80 | 0.88 | 0.86 | 0.917 |
Train accuracy | 0.71 | 0.83 | 0.77 | 0.83 | 0.978 |
Test accuracy | 0.71 | 0.83 | 0.81 | 0.81 | 0.975 |
Metastasis | Random Forest | Gradient Boosting | Deep Learning | ||
Text Feature (47) | AUC > 0.6 (17) | Text Feature (47) | AUC > 0.6 (17) | Train (37): Test (15) | |
Sensitivity | 0.77 | 0.80 | 0.76 | 0.85 | 0.958 |
Specificity | 0.74 | 0.66 | 0.76 | 0.73 | 0.990 |
AUC | 0.73 | 0.85 | 0.74 | 0.72 | 0.970 |
Train accuracy | 0.72 | 0.76 | 0.77 | 0.67 | 0.986 |
Test accuracy | 0.72 | 0.76 | 0.61 | 0.76 | 0.983 |