[파이썬] 윈도우 Graphviz를 사용하여 의사결정나무 시각화 하기

1. Graphviz 윈도우 버전 다운로드

2. 시스템 환경 변수 Path 편집

Graphviz파일의 bin폴더의 경로값을 입력해준다.

C:\Program Files\Graphviz\bin 대부분 이경로를 넣어주면 됩니다!

 

3. 쥬피터 노트북에 !pip install graphviz 입력

 

4. 그대로 입력

from graphviz import Source
from sklearn.tree import export_graphviz
Source(export_graphviz(model, feature_names=['sepal_length','sepal_width','petal_length','petal_width'],
                          class_names=['setosa','versicolor','virginica'],filled=True,rounded=True))

5. expot_graphviz의 인자값을 자신의 모델에 맞춰 수정해주시면됩니다

 

 

• model -> 자신이 학습이 완료된 의사결정나무로 만든 모델 이름을 넣어줍니다.
• feature_names= : 모델에 넣었던 x 변수 이름들을 리스트 형식으로 차례대로 넣어줍니다.
• class_names= : 클래스(정답라벨) 을 리스트형식으로 넣어줍니다
• filled= : True라면 그림들이 클래스 및 불순도에 맞춰 색칠을 해줍니다.
• rounded= : True라면 그림들의 모소리가 둥글게됩니다.

sklearn.tree.export_graphviz

 

 

 

 

 

 

 

 

 

 

 

 

 

※

본 게시물은 개인적인 용도로 작성된 게시물이며. 이후 포트폴리오로 사용될 정리 자료이니
불펌과 무단도용은 하지 말아주시길 부탁드리고, 개인 공부 목적으로 이용해주시면 감사하겠습니다.

※