问题背景及情况
目前想要将一个回归决策树的python2代码升级到python3.9,python2中回归决策树输出图片正常
,如下图所示:
现在升级到python3.9后,代码可以正常运行,也可以输出决策树的图片,但图片上出现一个黑块
fname = []
#fname.append(TREE_all_var[k].encode('utf-8'))
fname.append(TREE_all_var[k])
#print(fname.encode("raw_unicode_escape").decode("utf-8"))
dot_data1 = StringIO()
tree.export_graphviz(dtr,out_file=dot_data1, feature_names=fname, class_names = TREE_data,filled=True, rounded=True,
special_characters=True)
dot_data = dot_data1.getvalue()
dot_data = dot_data.replace('helvetica','"Microsoft Yahei"')
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_png("%s%s_%i_%s"%(outpath,TREE_all_var[k],maxdepth,filename[7]))
我的解答思路和尝试过的方法
尝试通过调整export_graphviz的参数来解决,但此法不通。
我想要达到的结果
希望输出的图片中不存在黑块。