决策树可视化中文乱码问题解决

最新推荐文章于 2022-11-27 14:08:52 发布

shangyf_418

最新推荐文章于 2022-11-27 14:08:52 发布

阅读量2k

点赞数 1

分类专栏： python 机器学习

本文链接：https://blog.csdn.net/shangyf_418/article/details/100893185

版权

通过使用notepad++修改tree.dot文件中fontname为'Microsoft YaHei'，然后在命令行用dot工具将.dot文件转换为PNG或PDF，可以有效解决决策树可视化过程中的中文乱码问题。

摘要由CSDN通过智能技术生成

# -------- 中文乱码问题解决 -------- 
''' 
# 先保存xxx.dot文件，然后打开保存的xxx.dot文件修改字体，修改为（注意两个fontname的位置）： 
digraph Tree { node [shape=box, style="filled, rounded", color="black", fontname="Microsoft YaHei"] ; 
edge [fontname="Microsoft YaHei"] ; 
'''

# 其余相关代码 -------------------------------------

# 需导入模块 
from sklearn import tree
import re
import pydotplus

# 1.直接生成图片（中文会乱码）
dot_data = tree.export_graphviz(rf_model, out_file=None,
                                feature_names=cols_name,
                                class_names=label_name,
                                filled=True, rounded=True,
                                special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
with open('1.png', 'wb') as f:
    f.write(graph.create_png())

#2. 生成tree.dot文件
with open("tree.dot",