设置决策树中节点拆分的条件

时间:2019-05-24 21:37:02

标签: machine-learning scikit-learn xgboost lightgbm

我想提供一个自定义指标,该指标应该进行优化以决定拆分时用于决策树,以替换标准的“基尼系数”。如何在任何决策树包中执行此操作。可以提升决策树。

编辑: 我想要实施一个标准,例如:

crit = (c1 + c2 - c3)/(2* sqrt(c2 + c4))

其中c1,c2,c3,c4是不同的类。这些类是不平衡的,因此我希望在计算时将其考虑在内(不要使用平衡的类权重)。

0 个答案:

没有答案