시작하기무료로 시작하기

RF 회귀 모델 평가하기

이전에 학습한 랜덤 포레스트 회귀 모델 rf의 테스트 세트 RMSE를 평가해 보세요.

데이터셋은 이미 전처리되어 80% 학습용, 20% 테스트용으로 분할되어 있습니다. 특성 행렬 X_test와 타깃 배열 y_test가 작업 공간에 준비되어 있습니다. 또한, 이전 연습 문제에서 학습한 모델 rf도 불러와 두었습니다.

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

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

강의 보기

연습 안내

  • sklearn.metrics에서 mean_squared_errorMSE라는 이름으로 임포트하세요.
  • 테스트 세트에 대한 예측값을 구해 y_pred에 저장하세요.
  • 테스트 세트의 RMSE를 계산하여 rmse_test에 저장하세요.

실습형 인터랙티브 연습

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

# Import mean_squared_error as MSE
from ____.____ import ____ as ____

# Predict the test set labels
y_pred = ____.____(____)

# Evaluate the test set RMSE
rmse_test = ____

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