【发布时间】:2019-07-21 06:12:01
【问题描述】:
我正在 python 中进行决策树回归。然而,对应于测试样本的预测目标值是该叶中目标变量的平均值。有没有一种方法,我们可以在那个桶中运行多元回归来获得测试样本的目标变量的估计值,而不是仅仅获得一个平均值?
P.S.:想知道 python 中的类似功能:https://www.researchgate.net/publication/2640479_Employing_Linear_Regression_in_Regression_Tree_Leaves
【问题讨论】:
-
我正在处理的数据在每片叶子中都有很高的可变性。因此,为了获得更好的预测,我试图在每棵树中定义一个多元回归。这将大大提高我的可预测性,而不是仅使用叶子的平均值作为预测
-
你在使用特定的库吗?
-
我正在使用 sklearn.tree 的 DecisionTreeRegressor
标签: python python-3.x scikit-learn regression decision-tree