Как извлечь структуру дерева XGBoost и оценки на каждом узле, а не только на листьях? - PullRequest
0 голосов
/ 19 марта 2020

Я использую Python. Делая

trained_xgb_model.dump_model("trees.txt")

, он может извлечь древовидную структуру и оценки. Текст выглядит так:

booster[0]:
0:[var1<10.2000008] yes=1,no=2,missing=2
    1:[var1<4.05000019] yes=3,no=4,missing=4
        3:leaf=0.148514852
        4:leaf=0.348837227
    2:leaf=-5.59405947
booster[1]:
0:[var1<10.2000008] yes=1,no=2,missing=2
    1:[var1<4.05000019] yes=3,no=4,missing=4
        3:leaf=0.133810416
        4:leaf=0.31406936
    2:leaf=-5.04019213

Однако он не дает балла для любого узла, который не является листом .

Мне нужно обрезать дерево. Мне нужна оценка на всех узлах. Ценю любое предложение.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...