【发布时间】:2015-06-18 23:13:05
【问题描述】:
我可以使用 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 中获取预测概率而不是类?
【问题讨论】:
-
发布reproducible example 会有所帮助。
forest的类是什么?您是否查看过该特定重载方法的帮助页面? -
感谢您的评论,我更新了问题。当然,我检查了 bigrf 类的 predict 方法,但找不到任何关于如何预测概率的提示。
-
不要编辑您的问题以包含“答案”。您应该通过单击响应旁边的复选标记来接受其他人发布的答案,或者如果没有其他答案适合您,请发布您自己的答案并接受它。