0

我正在使用 gbm 包创建一个简单的梯度提升机模型。我想计算一些预测与实际的统计数据(例如 GINI 比率,SSE)。我阅读了小插图和手册,但找不到从该程序的输出中获取预测值的位置。(当然我可以使用 gbm 在 train set 上进行预测,但是运行 .predict 需要很长时间)。

4

1 回答 1

0

也许这可以帮助你:

 pretty.gbm.tree(gbm1, i.tree=1) ## object gbm1 hast a list trees

 SplitVar SplitCodePred LeftNode RightNode MissingNode ErrorReduction Weight    Prediction
0        2  0.000000e+00        1         5           9      217.27619    250  2.074694e-04
1        1  1.232392e+00        2         3           4       30.95838     83 -5.554687e-03
2       -1 -7.445325e-03       -1        -1          -1        0.00000     60 -7.445325e-03
3       -1 -6.225881e-04       -1        -1          -1        0.00000     23 -6.225881e-04
4       -1 -5.554687e-03       -1        -1          -1        0.00000     83 -5.554687e-03
5        1  8.915513e-01        6         7           8       25.62508     89  5.689724e-03
6       -1  2.914792e-03       -1        -1          -1        0.00000     43  2.914792e-03
7       -1  8.283682e-03       -1        -1          -1        0.00000     46  8.283682e-03
8       -1  5.689724e-03       -1        -1          -1        0.00000     89  5.689724e-03
9       -1  8.360223e-05       -1        -1          -1        0.00000     78  8.360223e-05
于 2012-12-07T03:51:16.617 回答