二叉树的结构

Question

我想做什么：

我想展示一棵树。

我做了什么：

我曾 ML.net 使用随机 forrest 算法训练决策树模型。

随机福雷斯特算法产生了我要显示的多元回归树。

当我获取树列表时，我得到它们的格式如下：

leaf_num=20

split_feature=818 315 1707 1866 829 315 958 1682 1669 934 1794 1708 1274 1876 219 1892 557 38 755

split_gain=53.6443 26.8314 26.0605 25.312 24.9156 23.8195 26.0324 20.622 19.2891 19.2309 18.7427 17.5442 17.5224 16.147 15.3396 14.6873 14.3071 18.8269 14.2999
threshold=-0.001959169982001185 -0.0063232299871742717 -0.018274550326168534 -0.023883100599050518 -0.0033916500397026535 -0.0086862500756978971 0.0057975498493760833 -0.024933900684118267 0.025956150144338611 0.032694099470973022 -0.02218380011618137 0.018020399846136573 0.029140749946236614 -0.034148199483752244 0.0075621248688548812 0.02500609960407019 0.0055113600101321944 0.012193250004202129 0.018237499520182613

left_child=3 4 14 11 12 15 10 -4 13 -9 -7 -1 -2 -6 -3 -5 -12 -18 -8
right_child=1 2 7 5 8 6 18 9 -10 -11 16 -13 -14 -15 -16 -17 17 -19 -20

leaf_value=0.74074074074073915 0.34995047870584339 0.22257551669316322 -0.78431372549019063 -0.61827023271969384 0.6299212598425189 -0.45209903121636041 -0.68965517241379315 0.55182171913689448 0.59405940594059359 -0.78212290502792992 0.36304961678096054 -0.58823529411764608 -0.65789473684210453 -0.42461317020510986 -0.71428571428571386 0.19801980198019786 -0.3720106288751106 0.78853046594981979 0.031796502384737663
leaf_count=29 121 25 8 115 10 37 59 226 16 7 99 8 12 111 20 16 45 11 25
internal_value=0 0.963751 1.81717 -0.900576 0.0592242 -1.18872 -0.521437 2.3387 -1.13736 2.55618 0.291424 2.2605 1.29168 -1.68372 -0.974314 -2.59225 0.902294 -0.712251 -2.37643
internal_count=1000 556 286 444 270 407 276 241 137 233 192 37 133 121 45 131 155 56 84
shrinkage=0.2

ML.net的含class是RegressionTree.cs

根据我的理解 left_child 和 right_child 是 LteChild 和 GtChild，所有这些数组的长度都是 最大节点数 没有叶子（所以如果有 20 个叶子，长度是 19）。

split_feature是使用的特征的数据列。

我认为负值是叶子。

我的问题：

不幸的是，我无法像这样构建基于节点的树

     O
    / \
   O   O
  / \ / \
 L  L L  L

因为我不明白数组的值是如何排序的。

更新：

一如既往，当你向别人寻求帮助时，你自己很快就会得到另一个有前途的想法：

我深入研究了 ML.net 的代码，我想我找到了线索（感兴趣的人可以在第 856 行）。

我的猜测是必须有一个变量作为计数器，例如 int node

此计数器从 0

开始

要获得拆分列，需要获得 SplitFeatures[node]

子节点将是 LteChild[node] (LesserThanEqualChild) 和 GtChild[node] (GreaterThan)

等等。

更新2：

我想我做对了

结构化数据：

split_feature=  818 315 1707 1866 829 315 958 1682 1669 934 1794 1708 1274 1876 219 1892 557 38  755

                0   1   2    3    4   5   6   7    8    9   10   11   12   13   14  15   16  17  18
left_child=     3   4   14   11   12  15  10 -4    13  -9  -7   -1   -2   -6   -3  -5   -12 -18 -8
right_child=    1   2   7    5    8   6   18  9   -10  -11  16  -13  -14  -15  -16 -17   17 -19 -20

一旦我编写了一个解决方案，我就会 post 它作为答案。

Answer 1

我在 ML.net github 页面上发布了同样的问题，并从 Microsoft 的 Ivanidzo4ka 那里得到了很好的回答。

Link 问题：Link to github

二叉树的结构

Structure of a binary tree

c#

tree

regression

binary-tree

ml.net