

Abaconda Prompt 실행

pip install graphviz 입력

재부팅(주피터 노트북) -> 실행
from sklearn.tree import export_graphviz
export_graphviz(tree_model, out_file='tree.dot',
class_names=['독','식용'],
feature_names=X_one_hot.columns,
impurity=False,
filled=True)
import graphviz
with open('tree.dot', encoding='UTF8') as f:
dot_graph = f.read()
display(graphviz.Source(dot_graph))
