1. 학습
    2. /
  2. 강의
    3. /
  3. scikit-learn で学ぶ Supervised Learning
Connected

연습 문제

テストセットで予測する

前の演習では、線形回帰とリッジ回帰は同程度の結果に見えました。どちらのモデルを選んでも差し支えありませんが、テストセットでの予測性能を確認して、どちらかが優れているかを確かめてみましょう。

評価指標には root mean squared error（RMSE）を使います。2つのモデル名とインスタンスを含む辞書 models は、学習・テスト用配列 X_train_scaled、X_test_scaled、y_train、y_test とともにすでに読み込まれています。

지침

100 XP
  • root_mean_squared_error をインポートします。
  • スケーリング済みの学習用特徴量と学習用ラベルでモデルを学習させます。
  • スケーリング済みのテスト用特徴量で予測します。
  • テスト用ラベルと予測ラベルを渡して RMSE を計算します。