预测(随机森林):概率或投票对回归没有意义

时间:2014-06-04 09:25:09

标签: r random-forest predict

当我运行预测功能时,我收到此错误:

Error in predict.randomForest(fit, newdata = na.roughfix(csvTest[, -c(1:2,  : 
'prob' or 'vote' not meaningful for regression code here

这是代码:

fit <- foreach (ntree = rep (round(number_trees/nc), nc), 
            .combine = combine, .packages = 'randomForest') %dopar% randomForest(y=csv$mal,x=na.roughfix (csv[, c(1:2,4:5,8:13,17,19:20,22:29,ncol(csv))]), ntree = number_trees)
prob <- predict (fit, newdata = na.roughfix (csvTest[, -c(1:2,4:5,8:13,17,19:20,22:29,ncol(csvTest))]), type = 'prob')[, 2]

任何人都可以帮助我理解它的问题是什么吗? 感谢

1 个答案:

答案 0 :(得分:5)

我刚解决了。这是因为默认情况下我的目标变量(csv $ mal)被R设置为“数字”,而它应该是一个“因子”(0或1)。

使用数字,随机林运行type =“regression”,而不是type =“classification”。