如何在r中使用gbm获取预测

Question

fit <- gbm(Crop_Damage ~ Estimated_Insects_Count+Crop_Type+ Soil_Type
       +Pesticide_Use_Category+Number_Doses_Week+Number_Weeks_Used
       +Number_Weeks_Quit+Season,
       data = mydata, distribution="multinomial")

gbmpred <- predict(fit,mydata,n.trees = fit$n.trees)

我尝试了上面的代码，但它给了我概率。我想得到预测

Answer 1

默认情况下，预测函数的预测值为f（x），因此多项式（分类）案例的返回值将返回概率。您所要做的就是通过采用概率最高的类

将它们转换为响应