你需要使用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函数的参数更改为所需的格式。

import graphviz as graphvizimport numpy as npimport pandas as pdfrom sklearndatasets import load_irisfrom sklearnmodel_selection import train_test_splitfrom sklearntree import DecisionTreeClassifierfr

原文地址: http://www.cveoy.top/t/topic/4tg 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录