1. 학습
    2. /
  2. 강의
    3. /
  3. Pythonで学ぶ木ベースのMachine Learning
Connected

연습 문제

RF 回帰器を評価する

前の演習で学習したランダムフォレスト回帰器 rf のテストセット RMSE を評価します。

データセットはすでに前処理済みで、80% を学習用、20% をテスト用に分割しています。特徴量行列 X_test と配列 y_test はワークスペースで利用できます。さらに、前の演習で学習したモデル rf も読み込まれています。

지침

100 XP
  • sklearn.metrics から mean_squared_error を MSE としてインポートします。
  • テストセットのラベルを予測し、結果を y_pred に代入します。
  • テストセットの RMSE を計算し、rmse_test に代入します。