R~有没有办法从gbm中提取类预测？

Question

考虑代码：

require(rpart)
require(gbm)

fit = gbm(formula=Kyphosis~.,data=kyphosis,distribution="multinomial")
p   = predict.gbm(fit,kyphosis,n.trees=100)

结果：

有没有办法直接获得预测的类，例如C（不存在，不存在，...）

我可以使用以下代码执行此操作：

colnames(p)[apply(p,1,function(i){which(i==max(i))})]

结果：

[1]＆＃34;缺席＆＃34; ＆＃34;缺席＆＃34; ＆＃34;缺席＆＃34; ＆＃34;缺席＆＃34; ...

但是，这总是返回字符串。我希望返回的类型与预测器类型相同（例如，如果预测器是类型因子，我希望结果是因子，或者如果它是整数，结果应该是整数等等...） 。是否有任何返回预测类的默认函数？

所需行为的示例：

require(nnet)
fit = multinom(formula=Kyphosis~.,data=kyphosis)
predict(fit,kyphosis)