python 決定木グラフ

Question

pythonの決定木グラフの出力がうまくいかず困っております。
決定木自体はできております。

以下のようなエラーが出るのですが、解決方法ありますでしょうか。

import pydotplus
from sklearn.externals.six import StringIO
dot_data = StringIO()

from sklearn import tree

tr = tree.DecisionTreeClassifier(random_state=0)
tr.fit(X_train, Y_train)

tree.export_graphviz(tr, out_file="tree.dot")

graph = pydotplus.graph_from_dot_data(dot_data.getvalue())

（出力）
^
Expected {'graph' | 'digraph'} (at char 0), (line:1, col:1)

graph.write_pdf("graph.pdf")

（出力）

AttributeError Traceback (most recent call last)
<ipython-input-102-5f6f2882416d> in <module>()
----> 1 graph.write_pdf("graph.pdf")

AttributeError: 'NoneType' object has no attribute 'write_pdf'

よろしくお願いします。

Answer 1

sklearn.tree.export_graphviz — scikit-learn 0.19.1 documentation

Returns:

dot_data : string

String representation of the input tree in GraphViz dot format. Only returned if out_file is None.

らしいので、こう書き換えてみてください。

dot_str = tree.export_graphviz(tr, out_file=None)

graph = pydotplus.graph_from_dot_data(dot_str)

Comments

[yuyu127]

hayataka2049さん
ありがとうございます。

その後、「graph.write_pdf("graph.pdf")」とすると
→InvocationException: GraphViz's executables not found
というエラーが出てしまい、http://ruby.kyoto-wu.ac.jp/info-com/Softwares/Graphviz/を参考にしたのですが、うまくいきません。
解決方法はありますでしょうか。

[hayataka2049]

osと環境がわからないとなんともいえません。graphvizがうまく動いていないのは確実なので、その線でいろいろ整えてあげる必要があるみたい。
恐らくこの辺を参考にやれば上手くいくんじゃないですかね。ちょっと強引な方法みたいですけど
python3でscikit-learnの決定木を画像出力しようとしたら，graphviz's executablesエラーが出た - Qiita https://qiita.com/sh_tech_/items/8d3279c63fa98b21f879

[yuyu127]

hayataka2049さん
情報ありがとうございます。
このあたりの環境構築が苦手でうまくいってませんが、もう少し調べようと思います。