Machine learning algorithms able to predict the prognosis of gastric cancer patients treated with immune checkpoint inhibitors

Figure 2 Kaplan-Meier survival curve analysis of risk score based on XGBoost. A: When XGBoost was used to predict the patient’s overall survival, the Kaplan-Meier curve of the risk score in the training set; B: When XGBoost was used to predict the patient’s overall survival, the Kaplan-Meier curve of the risk score in the test set; C: When XGBoost was used to predict the patient’s progression-free survival, the Kaplan-Meier curve of the risk score in the training set; D: When XGBoost was used to predict the patient’s progression-free survival, the Kaplan-Meier curve of the risk score in the test set.