如何让我的 SHAP 图显示超过 20 个变量？

Question

如何让我的 SHAP 图在我的图表中显示超过 20 个变量。这是我的代码：

shap.initjs()
explainer = shap.TreeExplainer(model)
shap_values = explainer.shap_values(X_train)

shap.summary_plot(shap_values, X_train)
plt.savefig(Config.CLASH_PATH + '/plots/shap_' + target_cols + '.png')
plt.close()

SHAP graph

Answer 1

在 shap.summary_plot() 中使用选项 max_display=30。