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

时间:2016-01-29 20:26:44

标签: r machine-learning gbm

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)

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

1 个答案:

答案 0 :(得分:0)

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

将它们转换为响应