始める無料で始める

回帰木を評価する

この演習では、dt のテストセットにおける性能を Root Mean Squared Error(RMSE)で評価します。RMSE は、モデルの予測が実際のラベルと平均してどれくらい異なるかを表します。RMSE は、モデルの Mean Squared Error(MSE)の平方根を計算することで得られます。

特徴量行列 X_test、配列 y_test、そして前の演習で学習した決定木回帰器 dt は、ワークスペースに用意されています。

この演習はコースの一部です

Pythonで学ぶ木ベースのMachine Learning

コースを見る

演習の手順

  • sklearn.metrics から mean_squared_errorMSE としてインポートします。
  • テストセットのラベルを予測し、結果を y_pred に代入します。
  • MSE を呼び出してテストセットの MSE を計算し、結果を mse_dt に代入します。
  • テストセットの RMSE を計算し、rmse_dt に代入します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Import mean_squared_error from sklearn.metrics as MSE
from ____.____ import ____ as ____

# Compute y_pred
____ = ____.____(____)

# Compute mse_dt
____ = ____(____, ____)

# Compute rmse_dt
____ = ____

# Print rmse_dt
print("Test set RMSE of dt: {:.2f}".format(rmse_dt))
コードを編集して実行