考虑代码：
require(rpart)
require(gbm)
fit = gbm(formula=Kyphosis~.,data=kyphosis,distribution="multinomial")
p = predict.gbm(fit,kyphosis,n.trees=100)
结果：
有没有办法直接得到预测的类，例如 c(absent,absent,...)
我可以使用以下代码执行此操作：
colnames(p)[apply(p,1,function(i){which(i==max(i))})]
结果：
[1] “缺席” “缺席” “缺席” “缺席” ...
但是，这总是返回字符串。我希望返回的类型与预测器类型相同（例如，如果预测器是因子类型，我希望结果是因子，或者如果它是整数，结果应该是整数，等等...） . 是否有任何返回预测类的默认函数？
所需行为的示例：
require(nnet)
fit = multinom(formula=Kyphosis~.,data=kyphosis)
predict(fit,kyphosis)