在sklearn的DecisionTreeClassifier中访问非最佳拆分

时间:2019-09-06 02:42:08

标签: python scikit-learn decision-tree

我正在根据文档使用sklearn的DecisionTreeClassifier在虹膜数据上生成一棵树

https://scikit-learn.org/stable/modules/tree.html#tree

我想访问每个节点上次优分割的列表。也就是说,要看前n个分割,而不仅仅是前1个。
我只对具有唯一列名的前n个拆分感兴趣,因此如果n = 2,则前几个拆分为 petal_length <=2.45petal_length <= 2.3petal_width <= 1.75 我想看看 petal_length <=2.45petal_width <= 1.75

在我看来,我可能需要从头开始编写决策树并保存它们,或者在每个节点处放开最佳拆分列,并在该节点的行上训练一个拆分后的树,以获得下一个最佳拆分树分裂。 sklearn中有什么方法可以访问这些次优的分割?

预先感谢

0 个答案:

没有答案