Fig. 4
From: Application of machine learning in depression risk prediction for connective tissue diseases
SHAP summary plots for explaining the five variables contributing to the Catboost model. (a) The bar and dot chart displaying the feature importance rankings and the distribution of each variable influence on the Catboost outputs for the “none” category; (b) the bar and dot chart displaying the feature importance rankings and the distribution of each variable influence on the Catboost outputs for the “mild” category; (c) the bar and dot chart displaying the feature importance rankings and the distribution of each variable influence on the Catboost outputs for the “moderate and severe” category; (d) grouped-clustered heatmap illustrating a comparison of the five variables among the random ten CTD patients of the test cohort.
