I am using the multinomial distribution from the gbm
package in R. When I use the predict
function, I get a series of values:
5.086328 -4.738346 -8.492738 -5.980720 -4.351102 -4.738044 -3.220387 -4.732654
but I want to get the probability of each class occurring. How do I recover the probabilities? Thank You.
predict.gbm(..., type='response')
is not implemented for multinomial, or indeed any distribution other than bernoulli or poisson.So you have to find the most likely class (
apply(.., 1, which.max)
on the vector output from prediction), as desertnaut wrote:Just write a wrapper which accepts type='response' and returns this when it's a multinomial model.
Take a look at
?predict.gbm
, you'll see that there is a "type" parameter to the function. Try outpredict(<gbm object>, <new data>, type="response")
.