首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
社区首页 >问答首页 >Shap -在摘要图中不显示颜色条。

Shap -在摘要图中不显示颜色条。
EN

Stack Overflow用户
提问于 2021-12-23 03:48:24
回答 5查看 4.2K关注 0票数 11

当显示summary_plot时,颜色条不显示。

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
shap.summary_plot(shap_values, X_train)

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

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
shap.summary_plot(shap_values, X_train, plot_size=0.7)

下面是一个看起来不错的彩色条形图的例子。

有人知道这件事是否可以解决吗?

如何复制:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
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)

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

EN

回答 5

Stack Overflow用户

回答已采纳

发布于 2021-12-26 13:17:12

我遇到了和您一样的问题,我发现解决方案是将matplotlib降到3.4.3。看起来SHAP还没有针对matplotlib 3.5.1进行优化。

票数 5
EN

Stack Overflow用户

发布于 2022-03-10 09:07:29

我也遇到了类似的问题,我正在使用max_display=10。

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
plt.gcf().axes[-1].set_aspect(100)
plt.gcf().axes[-1].set_box_aspect(100)
票数 2
EN

Stack Overflow用户

发布于 2022-01-25 00:38:39

如前所述,在matplotlib.pyplot版本3.5中,颜色条或盒宽比的处理似乎已经改变。但是,您可以纠正这一点。

plt.gcf().axes[-1].set_aspect(1000)

  • Then

  • 使用shap.summary_plot(..., show=False)来允许改变上面提到的情节

  • ,用
  • 设置颜色条的方面,也可以设置颜色条的方框plt.gcf().axes[-1].set_box_aspect(1000)

的方面

这给了你以前的结果。如果您想使颜色条变厚,请将方面设置为100。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70461753

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
查看详情【社区公告】 技术创作特训营有奖征文