Структура двоичного дерева

Question

Что я пытаюсь сделать:

Я хочу отобразить дерево.

Что я сделал:

Я использовал ML.net для обучения модели дерева решений, используя алгоритм случайного Форреста.

Случайный алгоритм Форрест приводит к множеству деревьев регрессии, которые я хочу отобразить.

Когда я получаю список деревьев, я получаю их в следующем формате:

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_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] (Лучше чем)

и так далее.

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

как только я запрограммирую решение, я опубликую это как ответ.

Answer 1

Я разместил тот же вопрос на странице github ML.net и получил хороший ответ от Ivanidzo4ka из Microsoft.

Ссылка на выпуск: Ссылка на github