我在R中使用随机林包进行回归,它给出了两种信息:Mean of squared residuals
和%Var解释。但是我想计算训练和测试集的RMSE
和R^2
,任何人都可以帮助我如何找到这些信息?
答案 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
产生的对象,以及您可以在哪里找到各种性能指标。