Я использую 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
. Заранее спасибо!