我正在根据文档使用sklearn的DecisionTreeClassifier
在虹膜数据上生成一棵树
https://scikit-learn.org/stable/modules/tree.html#tree
我想访问每个节点上次优分割的列表。也就是说,要看前n个分割,而不仅仅是前1个。
我只对具有唯一列名的前n个拆分感兴趣,因此如果n = 2,则前几个拆分为
petal_length <=2.45
,
petal_length <= 2.3
,
petal_width <= 1.75
我想看看
petal_length <=2.45
,
petal_width <= 1.75
。
在我看来,我可能需要从头开始编写决策树并保存它们,或者在每个节点处放开最佳拆分列,并在该节点的行上训练一个拆分后的树,以获得下一个最佳拆分树分裂。 sklearn中有什么方法可以访问这些次优的分割?
预先感谢