1. 학습
    2. /
  2. 강의
    3. /
  3. Python으로 배우는 트리 기반 Machine Learning
Connected

연습 문제

GB 회귀 모델 학습하기

이제 이전 연습 문제에서 생성한 gradient boosting 회귀 모델 gb를 학습하고, 테스트 세트 레이블을 예측해 보겠습니다.

데이터셋은 학습 80%, 테스트 20%로 분할되어 있어요. 특성 행렬 X_train, X_test와 배열 y_train, y_test가 워크스페이스에 준비되어 있습니다. 또한, 이전 연습 문제에서 정의한 모델 인스턴스 gb도 미리 로드해 두었어요.

지침

100 XP
  • 학습 세트에 gb를 적합하세요.
  • 테스트 세트 레이블을 예측해 결과를 y_pred에 할당하세요.