[Python] Mac Graphviz 설치

의사결정나무를 시각화하기 위한 패키지인 Graphviz 설치 방법

 

Homebrew 설치

2022.09.15 - [Python] Mac에서 tesseract 설치하기 및 한글 추가

[Python] Mac에서 tesseract 설치하기 및 한글 추가

여기 앞부분 참고

 

 

터미널에서 graphviz 패키지 추가

brew install graphviz

 

 

pip 설치

pip install graphviz

 

 

 

from sklearn.tree import export_graphviz

export_graphviz(dt_clf, out_file = 'tree.dot', class_names=iris_data.target_names,
                feature_names=iris_data.feature_names, impurity=True, filled=True)

 

tree.dot라는 이름으로 시각화 파일 생성

 

 

시각화 파일 확인

import graphviz

with open('tree.dot') as f:
    dot_graph = f.read()
graphviz.Source(dot_graph)