1. 学ぶ
    2. /
  2. コース
    3. /
  3. scikit-learn による教師あり学習
Connected

演習

テストセットで予測する

前の演習では、線形回帰とリッジ回帰がほぼ同様の結果を示しました。どちらのモデルを選んでも問題ありませんが、テストセットで予測性能を確認することで、どちらが優れているかを判断できます。

評価指標には平均二乗誤差の平方根（RMSE）を使用します。2つのモデルの名前とインスタンスを格納した辞書 models、および学習・テスト用の配列 X_train_scaled、X_test_scaled、y_train、y_test はあらかじめ読み込まれています。

指示

100 XP
  • root_mean_squared_error をインポートしてください。
  • スケーリング済みの学習用特徴量とラベルを使って、モデルを学習させてください。
  • スケーリング済みのテスト用特徴量を使って予測を行ってください。
  • テストセットの正解ラベルと予測ラベルを渡して、RMSE を計算してください。