r - GBM多项分布，如何使用predict()得到预测类别？

Question

我正在使用 R 中的 gbm 包中的多项分布。当我使用 predict 函数时，我得到一系列值:

5.086328 -4.738346 -8.492738 -5.980720 -4.351102 -4.738044 -3.220387 -4.732654

但我想得到每个类别发生的概率。如何恢复概率？谢谢。

Answer 1

predict.gbm(..., type='response') 未针对多项式或除伯努利或泊松以外的任何分布实现。

因此，您必须找到最有可能的类(apply(.., 1, which.max) 在预测的向量输出上)，如 desertnaut wrote :

preds = predict(your_model, n.trees, newdata=...,type='response')

pred_class <- apply(preds, 1, which.max)

只需编写一个包装器，它接受 type='response' 并在它是多项模型时返回它。