1.安装可视化工具,graphviz。需要去官网下载电脑对应版本的exe进行安装。
网址:
2.添加环境变量
将下载的exe安装,并在系统环境变量中添加graphviz变量名,地址为graphviz安装路径中的bin/dot.exe
3.重启python环境
一定要重启环境。
4.绘图和保存方式
在python中添加如下代码
digraph = xgb.to_graphviz(model2) #model2是模型的名字
digraph.format = 'png' #保存为png的格式
digraph.view('D:/xgb_view2-'+name[i]) #保存路径
digraph2 = xgb.to_graphviz(model3)
digraph2.format = 'png'
digraph2.view('D:/xgb_view3-'+name[i])
5.用记事本方式打开xgb_view2文件,在里面添加一句中文字体格式的代码。
node [fontname="FangSong" shape=plaintext]
6.通过cmd进入xgb_view2的保存路径,用命令重新生成view图,起名为xgb_view2.jpg
dot -Tjpg xgb_view2 -o xgb_view2.jpg
7.效果![](https://i-blog.csdnimg.cn/blog_migrate/236beaccf386162777bb48d37029d5f5.jpeg)