在没有统计元数据的情况下渲染scikit-learn DecisionTreeClassifier?

时间:2017-08-03 20:58:15

标签: scikit-learn graphviz

使用sklearn.tree.export_graphiz()时,导出的树节点'标签包含统计元数据,例如samplesvalue

我希望在没有此数据的情况下向用户显示树的版本。也就是说,只有像length <= 7这样的决策标准和分类本身才会显示在节点内,没有别的。

export_graphviz()可以吗?我所看到的只是一个关键字选项,用于删除impurity分数,但没有其他统计信息。

1 个答案:

答案 0 :(得分:1)

目前代码中没有此类选项。我必须获取DOT文件内容的字符串(通过提供out_file=None参数),然后通过简单的DOT解析器处理它以从标签中删除所有元数据。