Я использую 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
Однако он не дает балла для любого узла, который не является листом .
Мне нужно обрезать дерево. Мне нужна оценка на всех узлах. Ценю любое предложение.