python graphviz sklearn决策树绘图解决中文乱码、修改分辨率

喝牛奶的草履虫

已于 2025-05-13 21:42:31 修改

阅读量740

点赞数 2

分类专栏： sklearn 文章标签： python sklearn 决策树 graphviz 中文分辨率

于 2024-04-04 15:17:19 首次发布

本文链接：https://blog.youkuaiyun.com/milk_paramecium/article/details/137375294

版权

sklearn 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了如何使用sklearn库中的`export_graphviz`函数生成决策树的DOT格式字符串，并进行字体替换和dpi设置，最后将其转换为图像文件并保存。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

修改sklearn导出的DOT格式的字符串

from sklearn import tree
import graphviz

dot_data = tree.export_graphviz(
	clf,
	out_file=None,
	feature_names=[], # 特征名称
	class_names=[], # 类别名称
	filled=True,
	rounded=True,
)
dot_data = dot_data.replace("helvetica", "Microsoft YaHei") # 修改中文字体
dot_data = dot_data[:-1] + 'graph [dpi=600];\n}'  # 修改dpi
# print(dot_data)
graph = graphviz.Source(dot_data, format="jpg")
print(graph)
graph.view("./imgs/tree") # 保存