二叉树的结构

时间:2018-07-12 09:20:16

标签: c# tree regression binary-tree ml.net

我要做什么:

我想显示一棵树。

我做了什么:

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

随机的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的包含类为RegressionTree.cs

根据我的理解,left_childright_child分别是LteChildGtChild,所有这些数组的最大节点数长度没有叶子(因此,如果有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)

以此类推。

UPDATE2:

我认为我没错

结构化数据:

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

一旦我编写了解决方案,我就会将其发布为答案。

1 个答案:

答案 0 :(得分:0)

我在ML.net github页面上发布了相同的问题,并从Microsoft的Ivanidzo4ka得到了很好的答案。

问题链接:Link to github

相关问题