sklean中决策树的可视化

最新推荐文章于 2023-08-12 12:12:41 发布

超人睡不着也变无奈

最新推荐文章于 2023-08-12 12:12:41 发布

阅读量166

点赞数

分类专栏：机器学习

本文链接：https://blog.csdn.net/XINGBAIDE/article/details/89152062

版权

机器学习专栏收录该内容

13 篇文章 0 订阅

订阅专栏

from sklearn.datasets import load_iris
from sklearn import tree
import graphviz
import sys
import pydotplus


def visualize(clf,path):
	dot_data = tree.export_graphviz(clf, out_file=None,feature_names=get_feature_names("./data/csv/E17.csv"))
    #dot_data = tree.export_graphviz(clf, out_file=None)
    graph = pydotplus.graph_from_dot_data(dot_data)
    graph.write_pdf(path+"iris.pdf")


iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
visualize(clf,"./output/")

with open("./output/iris.dot", 'w') as f:
    tree.export_graphviz(clf, out_file=f)