1. 学ぶ
    2. /
  2. コース
    3. /
  3. scikit-learn으로 배우는 Supervised Learning
Connected

演習

테스트 세트에 대한 예측

이전 연습 문제에서 선형 회귀와 ridge가 비슷한 결과를 보였어요. 두 모델 중 하나를 선택해도 무방하지만, 테스트 세트에서의 예측 성능을 확인해 어느 쪽이 더 나은지 비교해 볼 수 있어요.

이번에는 평가 지표로 RMSE(root mean squared error)를 사용해요. 두 모델의 이름과 인스턴스를 담은 딕셔너리 models가, 그리고 학습/타깃 배열 X_train_scaled, X_test_scaled, y_train, y_test가 미리 로드되어 있어요.

指示

100 XP
  • root_mean_squared_error를 임포트하세요.
  • 스케일링된 학습 특성과 학습 레이블로 모델을 학습하세요.
  • 스케일링된 테스트 특성으로 예측값을 만드세요.
  • 테스트 세트 레이블과 예측 레이블을 전달해 RMSE를 계산하세요.