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

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()

您还可以结合上述解决方案以获得结果的最佳分辨率。

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