1. 학습
    2. /
  2. 강의
    3. /
  3. HR Analytics: Python으로 직원 이탈 예측하기
Connected

연습 문제

트리 내보내기

Decision Tree 분류 과제에서는 트리가 너무 깊게 자라면 과적합이 발생하기 쉽습니다. 학습용과 테스트용 데이터의 정확도를 비교해 보면, 결과에서 과적합이 나타난 것을 확인할 수 있습니다. 이는 트리 시각화에서도 알 수 있습니다.

이 연습에서는 결정 트리를 텍스트 문서로 내보낸 뒤, 그 파일을 시각화에 활용해 보겠습니다.

지침

100 XP
  • sklearn.tree 하위 모듈에서 export_graphviz() 함수를 가져오세요.
  • model을 학습 데이터에 학습시키세요.
  • 시각화를 tree.dot 파일로 내보내세요.