如何在sklearn中输出决策树数据

Question

是否有一种输出决策树数据的方法，例如作为字典？我知道sklearn.tree.export_graphviz，这是一个很棒的视觉效果，但是我想知道是否有一种输出方式，例如：

是这样的：

{0: 'samples': 100.0%, 'value': [0.53, 0.47],
 1: 'samples': 72.6%, 'value': [0.61, 0.39],
 2: 'samples': 27.4%, 'value': [0.38, 0.62]}

将数据保存为dictionary，DataFrame，array等将使分析变得更容易，而不仅仅是查看一些彩色框。我已经浏览了sklearn文档，但找不到任何东西。

谢谢

Answer 1

该here似乎有一个实现。在玩完它之后，我认为限制因素之一就是像这样的字典深度。请记住，像这样的树不是：

{node1:[data],
 node2:[data],
 node3:[data],
 etc}

它更像是：

{node1:[data],
      sub-node1:{
                 sub-sub-node1:{
                                sub-sub-sub-node1:[data],sub-sub-sub-node2:[data],...}
                 sub-sub-node2:{
                                sub-sub-sub-node1:[data],sub-sub-sub-node2:[data],...}
                 etc}
      sub-node2:{etc}
}

如果您有一棵高（或None）max_depth

的树，谁会很快失控