wuchm 发表于 2015-3-23 15:10 
没明白哦,可否写个完整程序呢,将R平方添加到图中?
还有2个问题:
不好意思啊, 仔细一看我回答的是随机森林的.
一下子脑子抽住了.
总体上市很相似的。决策树的R-squared这么算:
(以下计算基于test set)
首先计算 SST = sum( (yi - ybar) ^ 2)
然后计算 每个叶节点k上的 Sk = sum( (yi - ybar) ^ 2), 把所有的Sk直接求和, 得到SSE
然后Rsquared就是1 - SSE / SST
我不知道你的数据形式是怎样的, 所以不知道怎么写啊。
其次,图像总会是可以改的啊。再不行可以根据结果自己写一个画图脚本嘛
但是你原来的图像是怎样的,最后的图像又是怎样的呢?