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

연습 문제

RF 회귀 모델 평가하기

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

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

지침

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