1. Învăţa
  2. /
  3. Courses
  4. /
  5. XGBoost로 익히는 Extreme Gradient Boosting

Connected

exercise

개별 XGBoost 트리 시각화하기

이제 XGBoost로 회귀와 분류 모델을 모두 구축하고 평가해 보았으니, 모델을 시각적으로 탐색하는 방법도 익혀 보겠습니다. 여기서는 전체 주택 데이터셋을 사용해 XGBoost가 만든 완전한 부스팅 모델에서 개별 트리를 시각화해 볼 거예요.

XGBoost에는 이런 시각화를 쉽게 해 주는 plot_tree() 함수가 있습니다. XGBoost 학습 API로 모델을 학습한 뒤, num_trees 인자를 사용해 그리고 싶은 트리 번호와 함께 모델을 plot_tree() 함수에 전달하면 됩니다.

Instrucţiuni

100 XP
  • "objective"는 "reg:squarederror", "max_depth"는 2로 하는 파라미터 딕셔너리를 만드세요.
  • 10회의 boosting 라운드와 위에서 만든 파라미터 딕셔너리를 사용해 모델을 학습하세요. 결과는 xg_reg에 저장하세요.
  • xgb.plot_tree()로 첫 번째 트리를 그리세요. 이 함수는 두 개의 인자를 받습니다. 모델(여기서는 xg_reg)과 0부터 시작하는 num_trees입니다. 따라서 첫 번째 트리를 그리려면 num_trees=0을 지정하세요.
  • 다섯 번째 트리를 그리세요.
  • 마지막(열 번째) 트리를 가로 방향으로 그리세요. 이를 위해 키워드 인자 rankdir="LR"을 추가로 지정하세요.