시작하기무료로 시작하기

GB 회귀 모델 평가하기

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

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

이 연습은 강의의 일부입니다

Python으로 배우는 트리 기반 Machine Learning

강의 보기

연습 안내

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

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

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

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

# Import mean_squared_error as MSE
____

# Compute MSE
mse_test = ____

# Compute RMSE
rmse_test = ____

# Print RMSE
print('Test set RMSE of gb: {:.3f}'.format(rmse_test))
코드 편집 및 실행