画出xgboost模型结构图
xgb.plot_tree(bst, num_trees=0, )
fig = pyplot.gcf()
fig.set_size_inches(120, 120)
fig.savefig('tree1.png')
bst是训练的模型,最后会把模型保存到tree1.png文件中
输出模型如下,xgboost中的模型特征会重复使用
画出xgboost模型结构图
xgb.plot_tree(bst, num_trees=0, )
fig = pyplot.gcf()
fig.set_size_inches(120, 120)
fig.savefig('tree1.png')
bst是训练的模型,最后会把模型保存到tree1.png文件中
输出模型如下,xgboost中的模型特征会重复使用