编辑 scikit-learn 决策树

Question

我想编辑 sklearn DecisionTree，例如更改条件或剪切节点/叶子等。

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

如何编辑决策树？

环境：

• Windows10
• python3.3
• sklearn 0.17.1

Answer 1

尽管文档说splitterkwarg forDecisionTreeClassifier是一个字符串，但您也可以给它一个类。证据：

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)