我想更改从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)