Граф решения дерева с pydotplus

Я пытаюсь создать графовую версию модели дерева решений scikit-learn. Моя цель - визуализировать его и в конечном итоге сохранить как файл.png (.jpeg тоже может работать). Основываясь на этом сайте - https://medium.com/@rnbrown/creating-and-visualizing-decision-trees-with-python-f8e8fa394176- - я написал код ниже, используя pydot. Это дает ошибку ниже, и мне интересно, если кто-нибудь знает, как решить эту проблему.

from sklearn.externals.six import StringIO  
from IPython.display import Image  
from sklearn.tree import export_graphviz
import pydotplus
dot_data = StringIO()
export_graphviz(dtree, out_file=dot_data,  
                filled=True, rounded=True,
                special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())  
Image(graph.create_png())

InvocationException                       Traceback (most recent call last)
<ipython-input-45-0305e90704a1> in <module>()
     10                 special_characters=True)
     11 graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
---> 12 Image(graph.create_png())

C:\ProgramData\Anaconda3\lib\site-packages\pydotplus\graphviz.py in <lambda>(f, prog)
   1795             self.__setattr__(
   1796                 'create_' + frmt,
-> 1797                 lambda f=frmt, prog=self.prog: self.create(format=f, prog=prog)
   1798             )
   1799             f = self.__dict__['create_' + frmt]

C:\ProgramData\Anaconda3\lib\site-packages\pydotplus\graphviz.py in create(self, prog, format)
   1958             if self.progs is None:
   1959                 raise InvocationException(
-> 1960                     'GraphViz\'s executables not found')
   1961 
   1962         if prog not in self.progs:

InvocationException: GraphViz's executables not found

0 ответов

Другие вопросы по тегам