当显示summary_plot时,颜色条不显示。
shap.summary_plot(shap_values, X_train)
我试过改变plot_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)
在这种情况下,显示的颜色条,但它非常小。我选择了这样一个例子,以便于检索数据。
发布于 2021-12-26 13:17:12
我遇到了和您一样的问题,我发现解决方案是将matplotlib降到3.4.3。看起来SHAP还没有针对matplotlib 3.5.1进行优化。
发布于 2022-03-10 09:07:29
我也遇到了类似的问题,我正在使用max_display=10。
plt.gcf().axes[-1].set_aspect(100)
plt.gcf().axes[-1].set_box_aspect(100)
发布于 2022-01-25 00:38:39
如前所述,在matplotlib.pyplot版本3.5中,颜色条或盒宽比的处理似乎已经改变。但是,您可以纠正这一点。
plt.gcf().axes[-1].set_aspect(1000)
shap.summary_plot(..., show=False)
来允许改变上面提到的情节
plt.gcf().axes[-1].set_box_aspect(1000)
的方面
这给了你以前的结果。如果您想使颜色条变厚,请将方面设置为100。
https://stackoverflow.com/questions/70461753
复制相似问题