R中的随机森林包

时间:2015-04-21 14:14:03

标签: r random-forest mse

我在R中使用随机林包进行回归,它给出了两种信息:Mean of squared residuals和%Var解释。但是我想计算训练和测试集的RMSER^2,任何人都可以帮助我如何找到这些信息?

1 个答案:

答案 0 :(得分:2)

对不起,这不是具体的答案,但我没有足够的信心发表评论。

如果没有可重复的例子,很难说你如何得到你想要的东西。但是,如果您在xtest=的调用中使用ytest=randomForest参数(假设您使用的是" randomForest"包),那么您要找的是什么应该是生成的randomForest对象的一部分。您要查看的是生成的随机森林列表的test部分。

一个尝试的例子:

rf.results <- randomForest( whatever arguments )
rf.results$test$mse  # mse (maybe you can take the square root to get rmse)
rf.results$test$rsq  # pseudo-R2 for random forest

如果您加载了随机林包,则可以验证此信息,并使用?randomForest自行进行一些探索。 &#34;价值&#34;文档的一部分详细说明了调用randomForest产生的对象,以及您可以在哪里找到各种性能指标。