回答问题

显示摘要\绘图时,颜色条不显示。

shap.summary_plot(shap_values, X_train)

我试过改变情节_size。当情节较高时,会出现颜色条,但它非常小 - 看起来不应该。

shap.summary_plot(shap_values, X_train, plot_size=0.7)

这是一个外观正确的颜色条的示例。

有谁知道这是否可以以某种方式解决?

如何复制:

import pandas as pd
import shap
import sklearn
from sklearn.ensemble import RandomForestRegressor

# a classic housing price dataset
X,y = shap.datasets.boston()

# a simple linear model
model = RandomForestRegressor(max_depth=6, random_state=0, n_estimators=10)
model.fit(X, y)
shap_values = shap.TreeExplainer(model).shap_values(X)
shap.summary_plot(shap_values, X)

在这种情况下,会显示彩条,但它非常小。我选择了这样一个例子,以便于检索数据。

Answers

我遇到了和你一样的问题,我发现解决方案是将 matplotlib 降级到 3.4.3 .. 似乎 SHAP 还没有针对 matplotlib 3.5.1 进行优化。