更改 SHAP 图的纵横比答案

【问题标题】：Change aspect ratio of SHAP plots更改 SHAP 图的纵横比
【发布时间】：2021-09-28 04:10:44
【问题描述】：

我想更改从shap 库生成的绘图的纵横比。

下面的最小可重现示例图：

import numpy as np
import pandas as pd  
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
from sklearn.datasets import load_boston
import shap


boston = load_boston()
regr = pd.DataFrame(boston.data)
regr.columns = boston.feature_names
regr['MEDV'] = boston.target

X = regr.drop('MEDV', axis = 1)
Y = regr['MEDV']

fit = LinearRegression().fit(X, Y)

explainer = shap.LinearExplainer(fit, X, feature_dependence = 'independent')

shap_values = explainer.shap_values(X)
shap.summary_plot(shap_values, X)

我可以用这个来保存图：

fig = shap.summary_plot(shap_values, X, show = False)
plt.savefig('fig_tes1.svg', bbox_inches='tight',dpi=100)

但我无法更改纵横比，例如将宽高比设为 4:3。

我已经读到我应该能够

plt.gcf()

但对我来说，这只是创建了一个新的空白情节。

<Figure size 432x288 with 0 Axes>

【问题讨论】：

标签： python pandas matplotlib shap

【解决方案1】：

更新

使用plot_size参数：

shap.summary_plot(shap_values, X, plot_size=[8,6])
print(f'Size: {plt.gcf().get_size_inches()}')

# Output
Size: [8. 6.]

您可以使用set_size_inches修改图形的大小：

...
shap.summary_plot(shap_values, X)

# Add this code
print(f'Original size: {plt.gcf().get_size_inches()}')
w, _ = plt.gcf().get_size_inches()
plt.gcf().set_size_inches(w, w*3/4)
plt.tight_layout()
print(f'New size: {plt.gcf().get_size_inches()}')

plt.savefig('fig_tes1.svg', bbox_inches='tight',dpi=100)

输出：

Original size: [8.  6.7]
New size: [8. 6.]

注意：修改宽度可能比修改高度更好：

_, h = plt.gcf().get_size_inches()
plt.gcf().set_size_inches(h*4/3, h)

【讨论】：