DecisionTreeRegressor MAE 使用均值而不是中值？

Posted 2023-03-12

【中文标题】DecisionTreeRegressor MAE 使用均值而不是中值？

【英文标题】：DecisionTreeRegressor MAE using mean instead of median?

【发布时间】：2018-09-19 05:05:48

【问题描述】：

DecisionTreeRegressor 有一个 MAE 标准，它使用每个终端节点的中值计算 MAE。有没有办法用每个终端节点的平均值来折射它来计算 MAE？

regr = DecisionTreeRegressor(max_depth=2,  criterion='mae')

【参考方案1】：

正如here 所讨论的那样，它不支持开箱即用（决策树也是如此）。

正如那里提到的，您必须在 Cython here 中实现自定义标准，因为 Python 太慢了。或者更具体地说here，在树包中计算Median，并将其添加到字典CRITERIA_REG，here。