LGB决策树可视化

1.官网下载graphivz软件http://www.graphviz.org/download/如安装路径是D:/Graphviz2.38，将D:/Graphviz2.38/bin添加入系统环境变量PATH中
2.用python安装库graphviz，联网安装pip install graphviz ，离线安装先下载安装包，解压后找到setup.py文件，python setup.py install
3.用python安装库pydotplus，同上

#决策树的helloworld 使用决策树对iris数据集进行分类
from sklearn.datasets import load_iris
from sklearn import tree
import pydotplus
import graphviz
import os
import pandas as pd
os.environ["PATH"] += os.pathsep + 'D:/Graphviz2.38/bin'  #注意修改你的路径
#导入iris数据集
iris = load_iris()
#初始化DecisionTreeClassifier
clf = tree.DecisionTreeClassifier()
#适配数据
clf = clf.fit(iris.data, iris.target)
#将决策树以pdf格式可视化
dot_data = tree.export_graphviz(clf, out_file=None)
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_pdf("iris.pdf")

若运行上述代码输出pdf，则说明配置正确。

#LGB树展示
from sklearn.datasets import load_iris
from sklearn import tree
import pydotplus
import graphviz
import os
import pandas as pd
os.environ["PATH"] += os.pathsep + 'D:/Graphviz2.38/bin'  #注意修改你的路径
import lightgbm as lgb
model=lgb.LGBMClassifier()
model.fit(iris.data,iris.target)
dot_data=lgb.create_tree_digraph(model,tree_index=0)
dot_data.format='PDF'
dot_data.render('lgb_iris_0.pdf')

中文显示乱码解决：

1.修改Graphviz配置文件字体，配置文件 fonts.conf 路径：D:\Graphviz2.38\fonts\fontd.conf

2.在dot_data文件digraph中加入
node[fontname=”fangsong” shape=plaintext]
3.在cmd窗口下输入
dot -Tpdf dot_data.dot -o test.pdf # 将结果保存为 PDF

模型文件构建决策树
children_left:[1,2,3,-1,-1,6,-1,-1,9,10,-1,-1,-1]
children_right:[8,5,4,-1,-1,7,-1,-1,12,11,-1,-1,-1]
默认根节点是0，-1代表叶子节点，0的子节点是1和8，1的子节点是2和5，2的子节点是3和4，5的子节点是6和7，8的子节点是9和12，9的子节点是10和11。根据children_left和children_right画树与可视化的树保持一致。