Skip to main content
. 2025 Jan 10;15:1605. doi: 10.1038/s41598-024-84702-8

Fig. 6.

Fig. 6

Analysis of the XGBoost model with TreeSHAP on the simulation test set for the SEL model with Inline graphic variables. (a) shows that the SEL covariate is considered, by far, to be the most important covariate in the model. The order of the other covariates remains the same when we train the model without our SEL variable (i.e. considered as not available). The Partial Dependence Plot in (b) depicts the relationship learnt by the model between the target variable Y and the SEL variable. We can see that it recovers the quadratic relationship between Inline graphic and Y as defined in Eq. (2).