为决策树模型绘制树时出现错误：“TypeError：‘模块’对象不可调用”

Question

我编写了代码来可视化决策树模型。最初，我遇到了诸如未找到 graphviz 的可执行文件之类的错误，但我将其路径添加到环境变量中，甚至重新安装了 graphviz 模块。现在似乎工作得很好。但现在出现以下错误：

  Traceback (most recent call last):
  File "C:/Ankur/Python36/Python Files/Decision_Tree.py", line 57, in 
  <module>
  Image(graph.create_png())
  TypeError: 'module' object is not callable

代码如下。

from sklearn import tree
from io import StringIO
from PIL import Image
from graphviz import Graph
import pydotplus as py
# Code for creating the model and fitting the data.
#...........................
dot_data=StringIO()
tree.export_graphviz(clf,out_file=dot_data)
graph=py.graph_from_dot_data(dot_data.getvalue())
Image(graph.create_png())

Answer 1

Image是一个模块，而不是一个函数。看：from PIL import Image

您可能打算这样做Image.open(graph.create_png())。