相关疑难解决方法(0)

显示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)

在这种情况下，会显示颜色条，但它很小。我选择这样的例子是为了方便检索数据。