来自 rpart 模型的 R 方

机器算法验证 r rpart
2022-03-29 21:05:53

如何从拟合rpart模型中提取 R 方?

rsq.rpart(fit)

绘制这两个图,但我只想提取整棵树的 R 平方值。

我认为这很明显,但大量搜索并没有真正提供任何有用的东西。您可以提供的任何帮助将不胜感激。

1个回答

R 的优势在于大多数时候您可以轻松访问源代码。所以在你的情况下,从

> rsq.rpart

(不带括号)查看函数的实际作用。R2值获得为

tmp <- printcp(fit)
rsq.val <- 1-tmp[,c(3,4)]  

对于每一行(又名,分割数),我们有“表观”和“相对”(wrt。交叉验证)统计信息。