트리 내보내기
Decision Tree 분류 과제에서는 트리가 너무 깊게 자라면 과적합이 발생하기 쉽습니다. 학습용과 테스트용 데이터의 정확도를 비교해 보면, 결과에서 과적합이 나타난 것을 확인할 수 있습니다. 이는 트리 시각화에서도 알 수 있습니다.
이 연습에서는 결정 트리를 텍스트 문서로 내보낸 뒤, 그 파일을 시각화에 활용해 보겠습니다.
이 연습은 강의의 일부입니다
HR Analytics: Python으로 직원 이탈 예측하기
연습 안내
sklearn.tree하위 모듈에서export_graphviz()함수를 가져오세요.model을 학습 데이터에 학습시키세요.- 시각화를
tree.dot파일로 내보내세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Import the graphical visualization export function
from sklearn.____ import export_graphviz
# Apply Decision Tree model to fit Features to the Target
model.____(features_train,target_train)
# Export the tree to a dot file
____(model,"tree.____")