2017-08-14 15 views
0

.pklファイルにモデルをエクスポートしました。今はJoblib経由でインポートしようとしていますが、これはsklearn.model_selection._search.GridSearchCVという形式でインポートします。.pklファイルからexport_graphvizを使用してモデルを可視化する

sklearn.treeインポートexport_graphvizexport_graphvizは、最初のパラメータとしてtree_を使用しているため、使用できません。

これを行う方法はありますか?ここに私のコードです:

export_graphviz(model,out_file="out.dot") Traceback (most recent call last): File "", line 1, in File "/home/anaconda3/lib/python3.6/site-packages/sklearn/tree/export.py", line 433, in export_graphviz recurse(decision_tree.tree_, 0, criterion=decision_tree.criterion) AttributeError: 'GridSearchCV' object has no attribute 'tree_'

答えて

0

どのようにしてgridSearchCVを定義して訓練しましたか? decision_treeがgridsearchcvオブジェクト

ある場合

decision_tree.best_estimator_.tree_ 

- :おそらくgridSearchCVでの基礎となる木がしてアクセスすることができ

関連する問題