Prognostic prediction models for postoperative patients with stage I to III colorectal cancer based on machine learning

Figure 3 Prediction of disease-free survival by machine learning models. The plots show the areas under the curve (AUCs) and their 95%CI. A: The linear regression (LR) model in the training set [class 1 (cl 1): Before: 0.77 [0.73-0.82), after: 0.75 (0.70-0.79); class 2 (cl 2): Before: 0.75 (0.71-0.80), after: 0.69 (0.63-0.74); class 3 (cl 3): Before: 0.90 (0.84-0.95), after: 0.87 (0.81-0.92)]; B: The LR model in the testing set [cl 1: Before: 0.71 (0.61-0.80), after: 0.70 (0.60-0.79); cl 2: Before: 0.70 (0.61-0.80), after: 0.69 (0.57-0.80); cl 3: Before: 0.65 (0.35-0.89), after: 0.68 (0.43-0.87)]; C: The linear discriminant analysis (LDA) model in the training set [cl 1: Before: 0.77 (0.73-0.82), after: 0.75 (0.70-0.80); cl 2: Before: 0.76 (0.71-0.80), after: 0.69 (0.63-0.75); cl 3: Before: 0.89 (0.83-0.95), after: 0.87 (0.81-0.92)]; D: The LDA model in the testing set [cl 1: Before: 0.71 (0.61-0.80), after: 0.70 (0.60-0.79); cl 2: Before: 0.70 (0.61-0.80), after: 0.69 (0.57-0.79); cl 3: Before: 0.62 (0.34-0.88), after: 0.69 (0.43-0.88)]; E: The eXtreme gradient boosting (XGBoost) model in the training set [cl 1: Before: 0.96 (0.95-0.98), after: 0.79 (0.75-0.83); cl 2: Before: 0.94 (0.92-0.96), after: 0.74 (0.69-0.79); cl 3: Before: 0.99 (0.97-1.00), after: 0.88 (0.84-0.92)]; F: The XGBoost model in the testing set [cl 1: Before: 0.69 (0.60-0.78), after: 0.71 (0.62-0.80); cl 2: Before: 0.66 (0.57-0.76), after: 0.72 (0.62-0.81); cl 3: Before: 0.70 (0.46-0.89), after: 0.70 (0.48-0.88)]; G: The categorical features and gradient boosting (CatBoost) model in the training set [cl 1: Before: 0.91 (0.88-0.93), after: 0.80 (0.75-0.84); cl 2: Before: 0.91 (0.88-0.93), after: 0.76 (0.71-0.81); cl 3: Before: 0.99 (0.97-1.00), after: 0.89 (0.85-0.93)]; H: The CatBoost model in the testing set [cl 1: Before: 0.75 (0.69-0.82), after: 0.73 (0.64-0.81); cl 2: Before: 0.71 (0.63-0.80), after: 0.72 (0.63-0.81); cl 3: Before: 0.68 (0.43-0.90), after: 0.70 (0.50-0.87)]. The curves of the models constructed with the full-variable datasets and the datasets containing only important variables are depicted with solid lines and dashed lines, respectively (abbreviated as “before” and “after” in this annotation).