相关疑难解决方法(0)

更改 SHAP 图的纵横比

我想更改从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)
Run Code Online (Sandbox Code Playgroud)

我可以使用以下方法保存该图:

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

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

我读到我应该能够

plt.gcf()
Run Code Online (Sandbox Code Playgroud)

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

<Figure size …
Run Code Online (Sandbox Code Playgroud)

python matplotlib pandas shap

3
推荐指数
1
解决办法
1万
查看次数

标签 统计

matplotlib ×1

pandas ×1

python ×1

shap ×1