Mac에 graphviz 설치하기

 

 

나는 결정 트리 알고리즘의 트리 규칙을 시각적으로 표현하고 싶어 설치하게 되었다.

 

Graphviz는 그래프 기반의 dot 파일로 기술된 다양한 이미지를 쉽게 시각화할 수 있는 패키지이다.

 

 

시작하기 전에 homebrew가 깔려있어야 한다.

2021/02/08 - [Mac] - Homebrew 설치하기

Homebrew 설치하기

1. graphviz 패키지 추가

brew install graphviz

 

2. 

pip install graphviz

 

끝 ! 

---

# export_graphviz() 함수 불러오기 
# graphviz가 읽어 들여서 그래프 형태로 시각화할 수 있는 출력 파일 생성

from sklearn.tree import export_graphviz

# export_graphviz()의 호출 결과로 out_file로 지정된 tree.dot 파일을 생성함
export_graphviz(dt_clf, out_file="tree.dot", class_names=iris_data.target_names,
               feature_names = iris_data.feature_names, impurity=True ,filled=True)
               
import graphviz
# 위에서 생성된 tree.dot 파일을 Graphviz가 읽어서 주피터 노트북상에서 시각화
with open("tree.dot") as f:
    dot_graph = f.read()
graphviz.Source(dot_graph)               

import graphviz 해주면 잘 실행되는 것을 볼 수 있다 !