【发布时间】:2015-10-19 04:53:34
【问题描述】:
我是 R 的新手,我正在努力创建我的第一个模型。我在一个 2 类随机森林项目中工作,到目前为止,我已经对模型进行了如下编程:
library(randomForest)
set.seed(2015)
randomforest <- randomForest(as.factor(goodkit) ~ ., data=training1, importance=TRUE,ntree=2000)
varImpPlot(randomforest)
prediction <- predict(randomforest, test,type='prob')
print(prediction)
我不确定为什么我没有得到模型的整体预测。我的代码中一定遗漏了一些东西。我得到了 OOB 和测试集中每个案例的预测,但没有得到模型的整体预测。
library(pROC)
auc <-roc(test$goodkit,prediction)
print(auc)
这根本不起作用。
我已经阅读了 pROC 手册,但我无法理解所有内容。如果任何人都可以帮助编写代码或发布一个好的实用示例的链接,那将非常有帮助。
【问题讨论】:
-
模型的“总体预测”究竟是什么?对教程链接的请求被认为是本网站的题外话。最好问一个明确的编程问题。
-
整体预测是指我的模型的预测分数。有关 AUC 代码的任何帮助/提示?
标签: r random-forest prediction roc auc