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

연습 문제

GB 회귀 모델 평가하기

이제 테스트 세트에 대한 예측값이 준비되었으니, 이를 사용해 gb의 테스트 세트 RMSE(Root Mean Squared Error)를 계산해 봅시다.

y_test와 예측값 y_pred는 작업 공간에 준비되어 있어요.

지침

100 XP

  • sklearn.metrics에서 mean_squared_error를 MSE라는 이름으로 임포트하세요.

  • 테스트 세트의 MSE를 계산해 mse_test에 할당하세요.

  • 테스트 세트의 RMSE를 계산해 rmse_test에 할당하세요.