如何解决决策树可视化的中文乱码问题?(部分解决)

最新推荐文章于 2024-04-04 15:17:19 发布

Sun_Sherry

最新推荐文章于 2024-04-04 15:17:19 发布

阅读量2k

点赞数 5

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： graphviz

本文链接：https://blog.youkuaiyun.com/yeshang_lady/article/details/120220778

Python 专栏收录该内容

57 篇文章

订阅专栏

在使用pydotplus和Graphviz进行决策树可视化时遇到中文乱码问题，传统解决方案针对旧版GraphViz无效。通过在原有可视化代码后添加设置fontname的代码，部分解决了乱码问题，但会导致异常。最终实现的解决方案是人为修改每个节点的fontname属性值为‘SimHei’，得到部分正确的可视化结果。

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

问题描述:在使用pydotplus和Graphviz对决策树结构进行可视化的时候，由于训练数据的特征名称使用的是中文，所以在最终的可视化结果中，中文被显示成了乱码，具体如下图：
在这里插入图片描述

从网上找到的一些资料说，需要对GraphViz中的font配置文件做修改，但这些资料中所说的解决方案都是针对GraphViz 2.39版本做的，对于Graphviz2.49(目前的最新版本)无效(因为里面的配置文件都变了，也有可能是我安装的Graphviz版本里面的包不全)。我的解决方案如下：

dot_data=tree.export_graphviz(dtc,
                              feature_names=feats,
                              class_names=['N','Y'],
                              filled=True,special_characters=True,
                              rounded=True)
graph=pydotplus.graph_from_dot_data(dot_data)
#在原有的可视化代码后面添加如下代码，人为修改每个节点的fontname属性值
for node in graph.get_nodes(): 
    node.set_fontname(lambda x:'SimHei')
Image(graph.create_png())