我正在尝试使用pydotplus可视化决策树。下面是代码片段。
if image_file_name is not None:
dot_data = export_graphviz(decision_tree_model, feature_names=features,out_file=None,filled=True,rounded=True)
graph = pydotplus.graph_from_dot_data(dot_data)
colors = ('turquoise', 'orange')
edges = collections.defaultdict(list)
for edge in graph.get_edge_list():
edges[edge.get_source()].append(int(edge.get_destination()))
for edge in edges:
edges[edge].sort()
for i in range(2):
dest = graph.get_node(str(edges[edge][i]))[0]
dest.set_fillcolor(colors[i])
graph.write_png(image_file_name)
无论何时执行最后一条语句,我都会收到此消息。
我执行的故障排除步骤:
使用的Python版本:3.7.1
使用的Anaconda版本:Anaconda3-5.3.1
有人可以从这里帮助我吗?