Предсказания внутренних узлов модели xgboost - PullRequest
0 голосов
/ 28 марта 2019

Можно ли рассчитать предсказания внутренних узлов модели xgboost?Пакет R, gbm, обеспечивает прогноз для внутренних узлов каждого дерева.

Выходные данные xgboost, однако, показывают только прогнозы для последних листьев модели.

Выходные данные xgboost:

Обратите внимание, что столбец «Качество» имеет окончательный прогноз для конечного узла в строке 6. Мне бы также хотелось, чтобы это значение было также для каждого из внутренних узлов.

   Tree Node  ID    Feature    Split  Yes   No Missing     Quality  Cover
1:    0    0 0-0 Sex=female  0.50000  0-1  0-2     0-1 246.6042790 222.75
2:    0    1 0-1        Age 13.00000  0-3  0-4     0-4  22.3424225 144.25
3:    0    2 0-2   Pclass=3  0.50000  0-5  0-6     0-5  60.1275253  78.50
4:    0    3 0-3      SibSp  2.50000  0-7  0-8     0-7  23.6302433   9.25
5:    0    4 0-4       Fare 26.26875  0-9 0-10     0-9  21.4425507 135.00
6:    0    5 0-5       Leaf       NA <NA> <NA>    <NA>   0.1747126  42.50

R gbm output:

В выводе пакета R gbm столбец прогнозирования содержит значения как для конечных узлов (SplitVar == -1), так и для внутренних узлов.Я хотел бы получить доступ к этим значениям из модели xgboost

   SplitVar SplitCodePred LeftNode RightNode MissingNode ErrorReduction Weight   Prediction
0         1   0.000000000        1         8          15      32.564591    445  0.001132514
1         2   9.500000000        2         3           7       3.844470    282 -0.085827382
2        -1   0.119585850       -1        -1          -1       0.000000     15  0.119585850
3         0   1.000000000        4         5           6       3.047926    207 -0.092846157
4        -1  -0.118731665       -1        -1          -1       0.000000    165 -0.118731665
5        -1   0.008846912       -1        -1          -1       0.000000     42  0.008846912
6        -1  -0.092846157       -1        -1          -1       0.000000    207 -0.092846157

Вопрос:

Как получить доступ или рассчитать прогнозы для внутренних узлов модели xgboost?Я хотел бы использовать их для жадной версии оценки SHAP для бедных.

...