import graphviz as graphvizimport numpy as npimport pandas as pdfrom sklearndatasets import load_irisfrom sklearnmodel_selection import train_test_splitfrom sklearntree import DecisionTreeClassifierfr
你需要使用Graphviz软件将mytree.dot文件转换为图片格式。你可以尝试以下代码:
import graphviz from sklearn.tree import export_graphviz
dot_data = export_graphviz(clf, out_file=None,
feature_names=feature_name,
class_names=['0', '1'],
filled=True, rounded=True,
special_characters=True)
graph = graphviz.Source(dot_data)
graph.render("iris")
这将生成一个名为iris的PDF文件,其中包含你的决策树图。如果你想生成PNG或其他格式的图像,请将render函数的参数更改为所需的格式。
原文地址: http://www.cveoy.top/t/topic/4tg 著作权归作者所有。请勿转载和采集!