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

Question

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

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

Answer 1

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

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