ROC曲线 - 模型性能误差

时间:2015-08-12 08:07:00

标签: r machine-learning random-forest roc

我正在尝试绘制一条ROC曲线来显示我的模型性能。使用randomForest

来拟合模型
prediction <- predict(fit, test, type="prob")
pred <- prediction(test$prediction, test$flag_cross_over )
pred2 <- prediction(abs(test$prediction + 
                    rnorm(length(test$prediction), 0, 0.1)), flag_cross_over)
perf <- performance( pred, "tpr", "fpr" )
perf2 <- performance(pred2, "tpr", "fpr")
plot( perf, colorize = TRUE)
plot(perf2, add = TRUE, colorize = TRUE)

所以使用测试数据我试图检查模型性能。因此,预测列保留所做的预测,flag_cross_over是模型的标签。

错误在于:

  

预测和标签的交叉验证运行次数必须相等。   

预测尺寸为410 2 dim的标签是410 1

我不确定为什么预测在维度中还有一个值

0 个答案:

没有答案