Graphviz笔记--主要是python与它的交互

关于windows 10 Python与Graphviz交互

首先应该准备工作有两个部分：

（1）首先安装好Graphviz链接

（2）设置系统环境环境Graphviz的环境

（3）设置dot的系统环境的设置，好了的时候在dos命令敲击dot -version 

from sklearn.datasets import load_iris
from sklearn import tree
import pydotplus
import os

iris = load_iris()

clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)

# 保存模型
with open("iris.dot", 'w') as f:
    f = tree.export_graphviz(clf, out_file=f)

# 画图，保存到pdf文件
# 设置图像参数
dot_data = 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)
# 保存图像到pdf文件
graph.write_pdf("iris.pdf")