R,gbm-package,如何获得因子类型的预测响应

时间:2016-10-11 14:52:07

标签: r gbm

美好的一天! 我在R中使用gbm-package。我希望得到因子类型中的预测值(在我的实例中:setosa,versicolor,virginika),而不需要对“预测”结果进行额外的改进。然后使用table()。但我明白了:

           setosa   versicolor    virginica
 [1,]  0.02670639 -0.008701904 -0.013467212
 [2,]  0.02670639 -0.006228872 -0.013467212
 [3,]  0.02670639 -0.008701904 -0.013467212

这是我的代码:

    library(gbm)
    attach(iris)
    data(iris)
    set.seed(1)
    train <- sample(1:nrow(iris), nrow(iris)/2)
    fit <- gbm(Species~., data=iris[train,], distribution="multinomial",n.trees = 5000, verbose=FALSE)
    predictions <- predict(fit, iris[-train,], n.trees = 5000) 
    predictions 

谢谢!

0 个答案:

没有答案