Создание участка pdp в Matplotlib - PullRequest
1 голос
/ 08 февраля 2020

Я использую pdpbox для создания графиков pdp для каждой из 50 функций в модели. Я успешно создал и сохранил эти графики, используя:

# pdp_plot (https://pdpbox.readthedocs.io/en/latest/pdp_plot.html)
for i, feat in enumerate(list_features):
    # increase i by 1 to help with name formatting
    n_feat = i + 1

    # pdp_isolate (https://pdpbox.readthedocs.io/en/latest/pdp_isolate.html)
    pdp_dove = pdp.pdp_isolate(model=model,
                               dataset=df_train,
                               model_features=list_features,
                               feature=feat)

    # generate plot
    fig, axes = pdp.pdp_plot(pdp_isolate_out=pdp_dove,
                             feature_name=feat,
                             plot_pts_dist=True)

    # save figure
    if n_feat < 10:
        n_feat = '0{0}'.format(n_feat)
    # save figure
    plt.savefig('{0}_pdp_{1}.png'.format(n_feat, feat))

Я мог бы остановиться здесь, если захочу, поскольку все графики в моем рабочем каталоге находятся в отформатированном и организованном виде. Тем не менее, мне интересно, можно ли было бы добавить каждый график в сетку подзаговоров с nrows=5 и ncols=10. Заранее спасибо!

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