1. 학습
    2. /
  2. 강의
    3. /
  3. scikit-learn으로 배우는 지도 학습
Connected

연습 문제

테스트 세트로 예측하기

앞선 연습 문제에서 선형 회귀와 릿지(Ridge)는 비슷한 결과를 보였습니다. 두 모델 중 어느 것을 선택해도 무방하지만, 테스트 세트에서의 예측 성능을 비교해 어느 모델이 더 우수한지 확인해 볼 수 있습니다.

평가 지표로는 평균 제곱근 오차(RMSE)를 사용합니다. 두 모델의 이름과 인스턴스를 담은 딕셔너리 models와 훈련 및 타깃 배열 X_train_scaled, X_test_scaled, y_train, y_test는 미리 로드되어 있습니다.

지침

100 XP
  • root_mean_squared_error를 임포트하세요.
  • 스케일링된 훈련 특성과 훈련 레이블을 사용해 모델을 학습시키세요.
  • 스케일링된 테스트 특성을 사용해 예측을 수행하세요.
  • 테스트 세트의 실제 레이블과 예측 레이블을 전달하여 RMSE를 계산하세요.