의사결정나무

Pywiki
둘러보기로 가기 검색하러 가기

1 개요[편집 | 원본 편집]

의사결정 나무를 만드는 가장 간편한 방법은 사이킷런을 이용하는 것.

2 사용법[편집 | 원본 편집]

의도 설명 방법
모델 설정
from sklearn import tree
classfier = tree.DeisionTreeClassifier(max_depth = 3)
모델 훈련
model = classifier.fit(X = 훈련데이터X, Y=훈련데이터Y)
모델로 예측
pred = model.predict(검사할데이터X)
모델 구조 살피기 나무 모델이 어떤 식으로 만들어졌는지 시각화.
tree.plot_tree(model,  # 살필 나무모델.
                proportion=True,  # 데이터가 분리된 비율 표시.
                filled=True,  # 색칠해서 데이터의 판단 확률을 시각화 할 수 있다.
                )