决策树二进制变量节点解释

时间:2019-10-04 12:37:53

标签: python machine-learning data-science decision-tree

我已经使用sklearn库在python中构建了决策树,并且遇到了一个问题,该问题如何解释二进制变量在节点中的拆分。看截图 here

因此,我有一个变量 if_successful ,该变量是二进制的,其中1表示事务成功,0表示事务不成功。该叶的标题显示 if_successful <= 0,002 。我该怎么解释? 我认为左侧有 True False ,所以左侧箭头 if_successful = 1 if_successful = 0 在右侧。但是,如果 if_successful <= 0,002 为True,那么基本上是 if_successful = 0 ?然后解释就相反了,对此我感到很困惑。 如何解释将标头拆分为二进制变量?

1 个答案:

答案 0 :(得分:0)

您的决策树将二进制变量视为数字变量,因此表示为if_successful <= 0.002。将变量转换为二进制或布尔值并训练模型,它将正常工作,并为您提供01拆分。