編集scikit-learn decisionTree

Question

sklearn decisionTree、 などを編集したいと思います。条件を変更したり、節点や葉を切り取ったりしてください。

しかし、ファイルにエクスポートしてインポートするには、編集する機能がないようです。

decisionTreeを編集するにはどうすればよいですか？

環境：

• Windows10
• python3.3
• sklearn 0.17.1

Answer 1

ドキュメントはDecisionTreeClassifierためsplitter kwargが文字列であることを言うにもかかわらず、あなたが与えることができますそれもクラスです。

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

そして実行します：証拠：あなたがここで見つけるスプリッタクラスの1サブクラス化できたよう

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

が見える

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

Answer 2

あなたが編集を考えている場合モデル、私はこれを行う簡単な方法はないと思います。ルールセット [1]、 [2]のエクスポート（視覚化）については議論がありましたが、ルールセットのインポートは行われていません。ただし、ルールセットを手動で編集しようとすると、最適な解決策が得られるのは何でしょうか。また、条件を本当に知っていれば、scikit-learnをまったく使用せずに、ネストされたif-else条件のセットを使うこともできます。

スプリッターのImplを変更する必要がある場合は、@ zemekenengとして提案することができます。