Блог

Я хотел бы изменить соотношение сторон графиков, созданных из 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>
 

Обновить

Использовать 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)