python使用graphviz工具画图
Posted Lenskit
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了python使用graphviz工具画图相关的知识，希望对你有一定的参考价值。
本示例数据集为鸢尾花数据集，运行后会生成两个文件，一个是iris，一个是iris.pdf。
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
from sklearn.tree import export_graphviz
import graphviz
iris = load_iris()
X = iris.data[:, 2:]
y = iris.target
tree_clf = DecisionTreeClassifier(max_depth=2)
tree_clf.fit(X, y)
dot_data = export_graphviz(tree_clf, out_file=None)
graph = graphviz.Source(dot_data)
graph.render("iris")
dot_data = export_graphviz(tree_clf, out_file=None,
feature_names=iris.feature_names[2:],
class_names=iris.target_names,
filled=True, rounded=True,
special_characters=True)
graph = graphviz.Source(dot_data)
以上是关于python使用graphviz工具画图的主要内容，如果未能解决你的问题，请参考以下文章