Posted at

Jupyter Notebook(iPython Notebook)内でscikit-learnの決定木(graphvizの.dotファイル)を表示する

More than 1 year has passed since last update.

pydotplusというライブラリを使うとできる

pip install pydotplus

import pydotplus

dot_data = StringIO()

tree.export_graphviz(
classifier,
out_file=dot_data,
feature_names=feature_names,
class_names=["False","True"],
filled=True,
rounded=True,
special_characters=True
)

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

いちいち、.dotファイルからpngファイルへの書き換えコマンドを打たなくてすむようになった