Python, пакет shap: как построить сетку графиков зависимости? - PullRequest
0 голосов
/ 22 октября 2019

Я пытаюсь построить сетку графиков зависимости из пакета shap. Вот код MWE для примера того, что я хочу:

fig, axs = plt.subplots(2,8, figsize=(16, 4), facecolor='w', edgecolor='k') # figsize=(width, height)
fig.subplots_adjust(hspace = .5, wspace=.001)

axs = axs.ravel()

for i in range(10):

    axs[i].contourf(np.random.rand(12,12),5,cmap=plt.cm.Oranges)
    axs[i].set_title(str(250+i))

plt.show()

An example layout of what I want

Вот код, который я имею до сих пор. Некоторые вещи не работают:

  1. На размер фигуры моей сетки не влияют мои figsize аргументы
  2. Мой код отображает большие версии моих графиков под сеткой.
  3. Только один из графиков зависимости отображается в сетке.
fig, axs = plt.subplots(1,8, figsize=(4, 2))
axs = axs.ravel()

for b in X_test.columns[:3]:
    for a in X_test.columns[:3]:
        shap.dependence_plot((a, b), shap_interaction_values, X_test)

Изображение того, что я получаю: What I am getting.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...