Pythonでのgraphvizを使った決定木のグラフのサイズを変更したいです。

Question

前提・実現したいこと

Pythonでのgraphvizを使った決定木のグラフのサイズを変更したいです。
サイズを変更する方法を教えていただきたいです。

発生している問題・エラーメッセージ

グラフのサイズが大きく画面からはみ出してしまいます。

該当のソースコード

export_graphviz(tree, out_file="tree.dot", class_names=["malignant", "benign"], 
                              feature_names=cancer.feature_names, impurity=False, filled=True)
with open("tree.dot") as f:
    dot_graph = f.read()
graphviz.Source(dot_graph)

試したこと

補足情報（FW/ツールのバージョンなど）

Answer 1

graphvizのjupyter連携の機能に描画を任せるとコードが面倒くさい上に勝手なサイズにされる上に調整が難しい気がするので、画像ファイルとして出力した上でjupyterの機能で読むと良いのでは？

export_graphviz(tree, out_file="tree.dot", class_names=["malignant", "benign"], 
                              feature_names=cancer.feature_names, impurity=False, filled=True)

from IPython.display import Image, display_png
!dot -Tpng tree.dot -o tree.png
display_png(Image("tree.png"))

sklearn.tree.export_graphviz — scikit-learn 0.20.2 documentation
jupyter notebookで画像を表示させたいだけならば、IPython.displayモジュールのメソッドを使うべし - Qiita

表示部分はMarkdownセルにして

<img src="tree.png">

とかでもいいかも。

Comments

[fuma3]

丁寧にありがとうございます！