python - 如何缩小绘图中的 XGB 子树并保存为图像？

Question

在训练了一个简单的 xgb 分类器模型后，我决定绘制树以查看拆分和节点。

我想看看如何减少子树的数量，使树更窄？另外，如何将其保存为 pgn/jpg？

谢谢！

#!pip install graphviz
#!apt-get install graphviz
from xgboost import plot_tree
plot_tree(model, num_trees=1, max_depth=1)
fig = plt.gcf()
fig.set_size_inches(30, 15)

标签： pythonmachine-learningxgboost