在Scikit的官网上 ,找到的生成可视化的决策树的代码如下:
from IPython.display import Image
import pydotplus
dot_data = sklearn.tree.export_graphviz(clf, out_file=None,
feature_names=iris.feature_names,
class_names=iris.target_names,
filled=True, rounded=True,
special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_pdf("iris.pdf")
运行报错如下:
pydotplus.graphviz.InvocationException: GraphViz’s executables not found
安装GraphViz可以解决。通过 brew install graphviz 命令安装。
安装完成后重启IDE ,就可以生成可视化的pdf和png了。