最近在学习决策树,然后用python使决策树可视化时出了点问题。Graphviz,matplotlib的注解等都可以使决策树可视化。
大多数版本是:
1.安装GraphViz
下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html,选择msi文件下载,安装即可。安装完成后,将bin目录加入到系统path环境变量中
2.下载 PyGraphviz 并安装,
pip 安装下载后的 whl 文件
pip install pygraphviz-1.3.1-cp34-none-win_amd64.whl
然后这时就出现了问题,因为python3.6版本没有pygraphviz ,如果你是2.7或者3.4的版本,可以安装这个。于是在网上又搜了搜,发现可以安装 pydotplus
pip install pydotplus 即可安装
下面是简单的代码:
from sklearn.datasets import load_iris from sklearn import tree from IPython.display import Image import pydotplus iris = load_iris() X = iris.data[:, 2:] y = iris.target tree_clf = tree.DecisionTreeClassifier(max_depth=2) tree_clf.fit(X, y) dot_data = \ tree.export_graphviz( tree_clf, out_file=None, feature_names=iris.feature_names[2:], class_names=iris.target_names, rounded=True, filled=True ) graph = pydotplus.graph_from_dot_data(dot_data) Image(graph.create_png()) #画出这个树 graph.write_png("tree1.png")
效果如下