pydotplus 工作不正常，决策树可视化错误？

Question

实际上我正在使用 sklearn 来可视化决策树。我已经安装了 graphviz 和 pydotplus 并且还设置了 graphviz 的环境变量，但是当我运行这段代码时，它给出了一个错误。

我正在使用“后凸畸形”数据集。

这是实际代码

from IPython.display import Image
from sklearn.externals.six import StringIO
from sklearn.tree import export_graphviz
import pydotplus

features = list(df.columns[1:])
features

Output:
['Age', 'Number', 'Start']

效果很好

但是当我运行下面的代码时。 代码：

dot_data = StringIO()  
export_graphviz(dtree, 
out_file=dot_data,feature_names=features,filled=True,rounded=True)

graph = pydotplus.graph_from_dot_data(dot_data.getvalue())  
Image(graph[0].create_png())  

显示错误。

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call 
last)
<ipython-input-44-28715c87781f> in <module>
      3 
      4 graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
----> 5 Image(graph[0].create_png())

TypeError: 'Dot' object is not subscriptable

Answer 1

尝试从Image(graph[0].create_png()) 中删除[0]。

它应该可以正常工作。

您将能够可视化您的决策树。