RF 회귀 모델 평가하기
이전에 학습한 랜덤 포레스트 회귀 모델 rf의 테스트 세트 RMSE를 평가해 보세요.
데이터셋은 이미 전처리되어 80% 학습용, 20% 테스트용으로 분할되어 있습니다. 특성 행렬 X_test와 타깃 배열 y_test가 작업 공간에 준비되어 있습니다. 또한, 이전 연습 문제에서 학습한 모델 rf도 불러와 두었습니다.
이 연습은 강의의 일부입니다
Python으로 배우는 트리 기반 Machine Learning
연습 안내
sklearn.metrics에서mean_squared_error를MSE라는 이름으로 임포트하세요.- 테스트 세트에 대한 예측값을 구해
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))