使用sklearn.tree.export_graphiz()
时,导出的树节点'标签包含统计元数据,例如samples
和value
。
我希望在没有此数据的情况下向用户显示树的版本。也就是说,只有像length <= 7
这样的决策标准和分类本身才会显示在节点内,没有别的。
export_graphviz()
可以吗?我所看到的只是一个关键字选项,用于删除impurity
分数,但没有其他统计信息。
答案 0 :(得分:1)
目前代码中没有此类选项。我必须获取DOT文件内容的字符串(通过提供out_file=None
参数),然后通过简单的DOT解析器处理它以从标签中删除所有元数据。