编辑 scikit-learn 决策树

Question

我想编辑 sklearn 决策树， 例如改变条件或切割节点/叶子等。

但似乎没有功能可以做到这一点，如果我可以导出到文件，请编辑它以导入。

如何编辑决策树？

环境：

• Windows10
• python3.3
• sklearn 0.17.1

Answer 1

即使文档说 DecisionTreeClassifier 的 splitter kwarg 是一个字符串，您也可以给它一个类。证据：

https://github.com/scikit-learn/scikit-learn/blob/412996f/sklearn/tree/tree.py#L353-L360

看起来您可以继承此处找到的 Splitter 类之一：

https://github.com/scikit-learn/scikit-learn/blob/0.17.X/sklearn/tree/_splitter.pyx

然后做：

my_decision_tree = sklearn.tree.DecisionTreeClassifier(splitter=mySplitter)

Answer 2

如果您正在考虑编辑模型，我认为没有简单的方法可以做到这一点。已经讨论了导出（相当可视化）规则集 [1], [2]，但没有讨论导入规则集。但是，当它适合最佳解决方案时，手动尝试编辑规则集有什么意义？再说一遍，如果你真的知道条件，你可以简单地使用一组嵌套的 if-else 条件，而根本不用 scikit-learn。

如果你需要改变分路器的Impl，你可以按照@zemekeneng的建议去做。