【问题标题】:Predict probabilities with bigrf使用 bigrf 预测概率
【发布时间】: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 方法,但找不到任何关于如何预测概率的提示。
  • 不要编辑您的问题以包含“答案”。您应该通过单击响应旁边的复选标记来接受其他人发布的答案,或者如果没有其他答案适合您,请发布您自己的答案并接受它。

标签: r bigdata


【解决方案1】:

根据this post,应该可以得到类概率

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

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

【讨论】:

  • 谁知道确定哪些列对应哪些类的一致方法?
  • attr(predictions@testvotes,"dimnames")$Class 似乎为您提供了一个类列表,但不能肯定它们与预测矩阵中的列并行索引
猜你喜欢
  • 2018-09-05
  • 1970-01-01
  • 2016-05-23
  • 1970-01-01
  • 2019-09-16
  • 1970-01-01
  • 2013-05-22
  • 1970-01-01
相关资源
最近更新 更多