美好的一天! 我在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
谢谢!