带有pydot的图决策树

Question

我正在尝试创建scikit学习决策树模型的图形版本并将其另存为.png文件（.jpeg也可以工作）。基于对此线程的建议-Visualizing decision tree in scikit-learn-我使用pydot编写了以下代码。它给出了下面的错误，我想知道是否有人知道如何解决这个问题。

import pydot
dotfile = StringIO()
tree.export_graphviz(dtree, out_file=dotfile)
pydot.graph_from_dot_data(dotfile.getvalue()).write_png("entropy_resampled.png")

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-47-bced3b678ff0> in <module>()
      4 dotfile = StringIO()
      5 tree.export_graphviz(dtree, out_file=dotfile)
----> 6 pydot.graph_from_dot_data(dotfile.getvalue()).write_png("entropy_resampled.png")

AttributeError: 'list' object has no attribute 'write_png'