export_graphviz()

 

 

 

export_graphviz()

·         {sklearn.tree}모듈의 export_graphviz() 메소드를 사용하여 의사결정트리를 시각화시켜줍니다.

·         export_graphviz() 메소드는 그래프 저장용 텍스트 파일 포맷인 .dot 파일포멧을 생성합니다.

 

 

# graphviz data 생성함수 임포트

from sklearn.tree import export_graphviz

sklearn.tree.export_graphviz(decision_tree, out_file=None, max_depth=None, feature_names=None, class_names=None, label='all', filled=False, leaves_parallel=False, impurity=True, node_ids=False, proportion=False, rotate=False, rounded=False, special_characters=False, precision=3)

 

 

parameter	description
decision_tree	decision tree classifier 의사결정나무 분류기(모델)
filled=True	bool, optional (default=False)   filled=True: True로 지정하면 노드의 클래스가 구분되도록 색으로 칠해집니다
rounded	bool, optional (default=False)   rounded=True: 둥근 노드상자와 Helvetica 폰트를 사용합니다.   When set to True, draw node boxes with rounded corners and use Helvetica fonts instead of Times-Roman.
class_names	list of strings, bool or None, optional (default=None) class 변수의 값 이름 설정   Names of each of the target classes in ascending numerical order.
feature_names	list of strings, optional (default=None) 컬럼이름들 지정   Names of each of the features.
out_file	file object or string, optional (default=None) 파일 저장 유무   out_file=None: 문자열인 dot tree포멧으로 표현   Handle or name of the output file. If None, the result is returned as a string.

 

 

https://scikit-learn.org/stable/modules/generated/sklearn.tree.export_graphviz.html