export_graphviz()

CODEDRAGON Development/Python

반응형



 

 

 

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

 


반응형

'Development > Python' 카테고리의 다른 글

graph_from_dot_data()  (0) 2020.03.08
learning_curve()  (0) 2020.03.07
UniversalBank.csv  (0) 2020.03.06
re.RegexObject 클래스  (0) 2020.03.06
바이트 코드  (0) 2020.03.05