2017-01-06 13 views
0

python scikitには、決定木をドットファイルにエクスポートするexport_graphvizというメソッドがあります。scikit-learnのファイルから決定木をロード

ドットファイルをscikitにインポートする方法があるのか​​どうかは、決定木として学びたいですか? sklearn.tree.import_graphviz()という関数のように?

+0

これは書きにくいでしょう。意思決定ツリーを再作成するために必要な情報は、「ドット」ファイルには取り込まれず、失われている可能性が非常に高いです。 "dot"ファイルはgraphvizをサポートするためのものであり、汎用ユーティリティであり、意思決定ツリーを表すように特別に設計されていません。ドキュメンテーションにも必要な機能はありませんでした。 – mikeTronix

+0

できません。 'graphviz'は単にあなたのツリーがどのように分割されたのかを示す画像表現です。 –

答えて

0

AFAIK簡単な方法はありません。 Graphvizは、決定木を視覚化するためにのみ使用できます。モデルを保存する場合は、Pickleを使用してモデルを保存します。例:

import cPickle 
# save the classifier 
with open('my_dumped_classifier.pkl', 'wb') as fid: 
    cPickle.dump(gnb, fid)  

# load it again 
with open('my_dumped_classifier.pkl', 'rb') as fid: 
    gnb_loaded = cPickle.load(fid)