决策树模型中节点的所有子节点都具有相同的预测

时间:2018-11-06 00:47:34

标签: apache-spark machine-learning decision-tree apache-spark-ml

我使用spark ML 2.3建立了DecisionTree分类器模型。这是模型的一部分。toDebugString

 Else (feature 1 > 0.25)
  If (feature 0 <= -3.3495)
   If (feature 1 <= 237.14999999999998)
    Predict: 3.0
   Else (feature 1 > 237.14999999999998)
    Predict: 3.0
  Else (feature 0 > -3.3495)
   Predict: 3.0

在其他节点的顶部,所有子节点都预测为3,那么为什么要有子节点,为什么不只有一个节点-像这样?

 Else (feature 1 > 0.25)
  Predict: 3.0

我刚刚设置了maxdepth参数。

设置mininfogain可以减少此类if / else节点的出现,但不会将其删除。我仍然得到if / else块都预测相同的值。

这是应该的样子还是我缺少什么?

0 个答案:

没有答案