(Scikit-learn) Graphviz를 통한 tree구조 시각화

환경설정

1. 다운로드
   https://graphviz.org/download/

• 설치시 자동으로 환경변수 설정됨

2. 라이브러리 설치

   !pip install graphviz

시각화

from sklearn.tree import export_graphviz
from graphviz import Source

graph = Source(export_graphviz(tree,
                              feature_names=data['feature_names'],
                              class_names=data['target_names'],
                              rounded=True,
                              filled=True))

graph