1. Learn
  2. /
  3. Kurser
  4. /
  5. HR Analytics: Python으로 직원 이탈 예측하기

Connected

övning

트리 내보내기

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

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

Instruktioner

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