使用 Scikit Learn DecisionTreeClassifier 保存 Graphviz 决策树图

Question

我能够使用图形可视化生成决策树图，但是当我尝试保存它时（即out_file）

我收到以下错误：

调用进程错误：命令"['点'， '-Tsvg']' 返回非零退出 状态 1.

这是我的代码：

# create plot for decision tree
dot_data = tree.export_graphviz(model, 
                                out_file='tree.dot', # this is what triggers the error
                                feature_names=X_test.columns,
                                class_names=['Active','Churned']) 
graph = graphviz.Source(dot_data) 
# view plot 
graph

out_file参数是触发错误的参数。我认为这应该是一个非常简单的解决方案，但我还没有从我的搜索中找到答案。

提前谢谢你。

Answer 1

您应该在 tree.export_graphviz（） 的末尾添加一个参数 "out_file = None"。