如何在形状摘要图上更改轴

Question

我正在遵循列出的代码，仅具有略有不同的值（因此我的代码完全如链接中所述）。

我的输出看起来像这样：

有人可以向我展示如何更改X轴（理想情况下是自动的，因为我必须为不同的数据集制作多个图），因此数据不那么堆积向上看到完整的情节？

Answer 1

关于问题中链接的代码，您可以在计算shap_values之后尝试以下解决方案：

import matplotlib.pyplot as plt 
.
.
# Calculate shap_values
fig = shap.summary_plot(shap_values[1], X_test, show=False)
_, h = plt.gcf().get_size_inches()
plt.gcf().set_size_inches(h*5, h)
plt.show()

或

import matplotlib.pyplot as plt 
.
.
# Calculate shap_values
shap.summary_plot(shap_values[1], X_test, show=False)
ax = plt.gca()

# You can change the min and max value of xaxis by changing the arguments of:
ax.set_xlim(-0.5, 0.5) 
plt.show()

您也可以将上述解决方案组合起来以获取该解决方案最佳解决方案。

请参阅此答案以获取更多信息。