1.windows安装Graphviz2.38
安装地址：https://graphviz.gitlab.io/_pages/Download/Download_windows.html
2.在python文件头添加这两行代码
import os os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/' # 其中C:/Program Files (x86)为安装地址，下载下来的是zip需要解压到这个位置
3.实例代码
def draw_xgboost(self, model):
"""
this function is able to draw xgboost tree
:param model: xgboost model
:return:
"""
self.ceate_feature_map(self.pre_data.cheat_data.columns.values.tolist()) # 特征名列表
fig, ax = plt.subplots()
fig.set_size_inches(60, 30)
xgboost.plot_tree(model, ax=ax, fmap='xgb.fmap')
fig.savefig('xgb_tree.jpg')
def ceate_feature_map(self, features):
"""
this function is able to create xgboost feature map
:param features: 数据列名
:return:
"""
outfile = open('xgb.fmap', 'w')
i = 0
for feat in features:
outfile.write('{0}\t{1}\tq\n'.format(i, feat))
i = i + 1
outfile.close()
转载于:https://www.cnblogs.com/callyblog/p/9391963.html
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请联系我们举报，一经查实，本站将立刻删除。
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/221182.html原文链接：https://javaforall.net
