Python決定木の可視化

Question

jupyterでpythonで木構造を可視化したいと考えています。Graphvizをインストールし、パスも通したのですが、以下のコードを入力しても'C:\Users\ユーザー名'と出てしまいます。
ご助言いただければ幸いです。

from sklearn import tree
import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files/Graphviz2.38/bin/'
import pydotplus
from IPython.display import Image
from graphviz import Digraph
from sklearn.externals.six import StringIO
dot_data = StringIO()
tree.export_graphviz(clf, out_file=dot_data,feature_names=X2.columns, max_depth=3)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_pdf("graph.pdf")
os.getcwd()

Answer 1

graph.write_pdfが作図のコードです。ただし、このコードはpdfファイルにすることを意味しているので、別途そのファイルを開く必要があります。
コードによると、ファイル名はgraph.pdfとなっています。ディレクトリは、指定がないのでカレントディレクトリになります。このカレントディレクトリは最後のコードで出力しているので、これは参考にすればいいでしょう

Comments

[kinos]

ありがとうございます。解決しました！

Answer 2

環境何でやってます？

google colaboratoryをお使いでしたら

# Pip install
!pip install pydotplus
!pip install graphviz
# Apt-get install
!apt-get install graphviz

jupyterNotebookなら

conda install graphviz
pip install graphviz

Comments

[kinos]

ありがとうございます。解決しました！