从xgboost图树中提取权重和树结构

Question

我在python中使用xgboost包，并在培训后提取树结构。例如，我将了解在每个节点上选择的特性和阈值，以便将树结构导出到函数。 另外，我还需要知道每棵树的重量。（我们知道，树的结果将组合成w1*h1+w2h2+…+wn*hn，其中wi是权重，hi是第i棵树在boosting中的答案）。实际上，我需要有权使用权重和内部树。我在分类。 我的另一个问题是，当我使用“from xgboost import plot_tree”时，我收到一个错误消息“package does not this function”。我怎样才能画出我的树？在

谢谢

Answer 1

您可以将xgboost模型转储到文本文件中，然后自己解析它。文件如下所示：

增压器[0（<；-树id）]： 0（<；-节点id）：[f317（<；-功能名称）<；0.187154]是=1（<；-子节点id），否=2，缺少=1 1： [f317<；0.071172]是=3，否=4，缺失=3 ... 6379:叶=0.125（<；-该叶的重量）

最后，它是所有叶子的加权和。这是二进制分类和回归的情况。在