微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

不使用 Sklearn 或 PPrint 包打印决策树的输出

如何解决不使用 Sklearn 或 PPrint 包打印决策树的输出

我正在尝试以以下格式打印决策树的输出enter image description here

我将决策树存储为嵌套字典。我正在尝试对字典使用嵌套循环,但没有结果。

enter image description here

请让我知道有关如何实现这一目标的任何想法??

输出必须是这样的

|--- XO <= 0.50
|   |--- XM <= 0.50
|   |   |--- XF <= 0.50
|   |   |   |--- class: 0
|   |   |--- XF >  0.50
|   |   |   |--- class: 0
|   |--- XM >  0.50
|   |   |--- XB <= 0.50
|   |   |   |--- XF <= 0.50
|   |   |   |   |--- XG <= 0.50
|   |   |   |   |   |--- class: 0
|   |   |   |   |--- XG >  0.50
|   |   |   |   |   |--- XD <= 0.50
|   |   |   |   |   |   |--- class: 1
|   |   |   |   |   |--- XD >  0.50
|   |   |   |   |   |   |--- class: 0
|   |   |   |--- XF >  0.50
|   |   |   |   |--- class: 1
|   |   |--- XB >  0.50
|   |   |   |--- XI <= 0.50
|   |   |   |   |--- class: 0
|   |   |   |--- XI >  0.5

解决方法

如果您不知道必须循环遍历的嵌套字典的数量,可以使用递归:

def dictionary_print(dict_):
    for key,values in dict_.items():
        if isinstance(values,dict): dictionary_print(values)
    else: print(key,':',values)

此代码来自this post

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。