r语言 - 使用 bigrf 预测概率

Question

我能够用bigrf()包构建一个模型，但是有没有办法预测概率而不是类？对于类预测，我使用

predictions <- predict(forest, test, testset$y)

森林是一个模型。我尝试了type = "prob"但没有做任何事情。有没有办法做到这一点？

我有大数据，所以我需要使用这个包才能处理它。

上级：

library(bigrf)
library(randomForest)
data("iris")
iris <- iris[iris$Species != "virginica",]
x <- iris[,1:4]
y <- iris$Species
vars <- c(1:4)
s = sample(1:nrow(x), 60)
registerDoParallel(cores=detectCores(all.tests=TRUE))
forest <- bigrfc(x[s, ], y[s], ntree=5L, varselect=vars)
predictions <- predict(forest, x[-s, ])

那么，问题是如何从对象类 bigrfc 中获取预测中的概率而不是类？

Answer 1

根据这篇文章，应该可以通过以下公式获得类概率

predictions_probs <- predictions@testvotes/rowSums(predictions@testvotes)

不过我还没有测试过。呵呵。