В настоящее время я работаю над проблемой классификации и хочу создать визуализации, имеющие важное значение. Я использую пакет Python XGBoost, который уже предоставляет графики важности функций. Тем не менее, я нашел shap (https://github.com/slundberg/shap),) библиотеку Python, которая создает очень хорошие графики важности объектов на основе древовидных классификаторов. Все работает отлично, однако я также могу сохранить созданные графики как PNG, если я попытаюсь сохранить их в виде PDF или SVG я получаю исключение. Вот что я делаю:
Сначала я обучаю модель XGBoost и получаю модель, обозначенную bst.
train = remove_labels_for_binary_df(dataset_fc_baseline_1[0].train)
test = remove_labels_for_binary_df(dataset_fc_baseline_1[0].test)
results, bst = xgboost_with_bst(*transform_feat_to_num(train, test))
Затем я создаю значения формы, использую их для создания сводного графика и сохраняю визуализацию создания. Все отлично работает, если я сохраню сюжет как plt.savefig('shap.png').
import shap
import matplotlib.pyplot as plt
shap.initjs()
explainer = shap.TreeExplainer(bst)
shap_values = explainer.shap_values(train)
fig = shap.summary_plot(shap_values, train, show=False)
plt.savefig('shap.png')
Однако мне нужны графики PDF или SVG вместо png, и поэтому я попытался сохранить их с помощью plt.savefig('shap.pdf'), который обычно работает нормально, но выдает следующее исключение для графика shap.
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-39-49d17973f438> in <module>()
1 fig = shap.summary_plot(shap_values, train, show=False)
----> 2 plt.savefig('shap.pdf')
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\pyplot.py in
savefig(*args, **kwargs)
708 def savefig(*args, **kwargs):
709 fig = gcf()
--> 710 res = fig.savefig(*args, **kwargs)
711 fig.canvas.draw_idle() # need this if 'transparent=True' to reset
colors
712 return res
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\figure.py in
savefig(self, fname, **kwargs)
2033 self.set_frameon(frameon)
2034
-> 2035 self.canvas.print_figure(fname, **kwargs)
2036
2037 if frameon:
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\backend_bases.py in
print_figure(self, filename, dpi, facecolor, edgecolor, orientation, format,
**kwargs)
2261 orientation=orientation,
2262 bbox_inches_restore=_bbox_inches_restore,
-> 2263 **kwargs)
2264 finally:
2265 if bbox_inches and restore_bbox:
C:\Users\Studio\Anaconda3\lib\site-
packages\matplotlib\backends\backend_pdf.py in print_pdf(self, filename,
**kwargs)
2584 RendererPdf(file, image_dpi, height, width),
2585 bbox_inches_restore=_bbox_inches_restore)
-> 2586 self.figure.draw(renderer)
2587 renderer.finalize()
2588 if not isinstance(filename, PdfPages):
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\artist.py in
draw_wrapper(artist, renderer, *args, **kwargs)
53 renderer.start_filter()
54
---> 55 return draw(artist, renderer, *args, **kwargs)
56 finally:
57 if artist.get_agg_filter() is not None:
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\figure.py in
draw(self, renderer)
1473
1474 mimage._draw_list_compositing_images(
-> 1475 renderer, self, artists, self.suppressComposite)
1476
1477 renderer.close_group('figure')
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\image.py in
_draw_list_compositing_images(renderer, parent, artists, suppress_composite)
139 if not_composite or not has_images:
140 for a in artists:
--> 141 a.draw(renderer)
142 else:
143 # Composite any adjacent images together
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\artist.py in
draw_wrapper(artist, renderer, *args, **kwargs)
53 renderer.start_filter()
54
---> 55 return draw(artist, renderer, *args, **kwargs)
56 finally:
57 if artist.get_agg_filter() is not None:
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\axes\_base.py in
draw(self, renderer, inframe)
2605 renderer.stop_rasterizing()
2606
-> 2607 mimage._draw_list_compositing_images(renderer, self,
artists)
2608
2609 renderer.close_group('axes')
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\image.py in
_draw_list_compositing_images(renderer, parent, artists, suppress_composite)
139 if not_composite or not has_images:
140 for a in artists:
--> 141 a.draw(renderer)
142 else:
143 # Composite any adjacent images together
C:\Users\Studio\Anaconda3\lib\site-packages\matplotlib\artist.py in
draw_wrapper(artist, renderer, *args, **kwargs)
58 renderer.stop_filter(artist.get_agg_filter())
59 if artist.get_rasterized():
---> 60 renderer.stop_rasterizing()
61
62 draw_wrapper._supports_rasterization = True
C:\Users\Studio\Anaconda3\lib\site-
packages\matplotlib\backends\backend_mixed.py in stop_rasterizing(self)
128
129 height = self._height * self.dpi
--> 130 buffer, bounds =
self._raster_renderer.tostring_rgba_minimized()
131 l, b, w, h = bounds
132 if w > 0 and h > 0:
C:\Users\Studio\Anaconda3\lib\site-
packages\matplotlib\backends\backend_agg.py in tostring_rgba_minimized(self)
138 [extents[0] + extents[2], self.height - extents[1]]]
139 region = self.copy_from_bbox(bbox)
--> 140 return np.array(region), extents
141
142 def draw_path(self, gc, path, transform, rgbFace=None):
ValueError: negative dimensions are not allowed
У вас есть идеи, как это исправить?
Заранее спасибо!