安装graphviz
从以下链接下载GraphViz:https://www.php.cn/link/bfaa2ace1082af90074d02006690ddb2。
通过命令conda install python-graphviz来安装graphviz的Python库。
生成可视化文件的代码如下,使用JavaScript语言:
import graphviz // 忽略测试 from sklearn import tree print(data.columns) // ID3为决策树分类器,fit之后得到的模型,注意必须在fit之后执行,predict之后运行会报错 dot_data = tree.export_graphviz(ID3, out_file=None, feature_names=data.columns[:-1], class_names=np.unique(y)) // 忽略测试 graph = graphviz.Source(dot_data) // 忽略测试 // 在同级目录下生成tree.pdf文件 graph.render("tree") // 忽略测试
立即学习“Python免费学习笔记(深入)”;
参考资料:https://www.php.cn/link/2cfc7c2a1eabaff9fdcc09b009e8edad