始める無料で始める

線形回帰 vs 回帰木

この演習では、dt のテストセットRMSEを、線形回帰モデルで達成した値と比較します。すでに線形回帰モデル lr をインスタンス化し、dt と同じデータセットで学習済みです。

特徴量行列 X_test、ラベル配列 y_test、学習済みの線形回帰モデル lr、エイリアス MSE でインポートされた mean_squared_error 関数、そして前の演習で計算した rmse_dt がワークスペースに用意されています。

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

Pythonで学ぶ木ベースのMachine Learning

コースを見る

演習の手順

  • 線形回帰モデル(lr)を使ってテストセットのラベルを予測し、結果を y_pred_lr に代入します。

  • テストセットのMSEを計算し、結果を mse_lr に代入します。

  • テストセットのRMSEを計算し、結果を rmse_lr に代入します。

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

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

# Predict test set labels 
____ = ____.____(____)

# Compute mse_lr
____ = ____(____, ____)

# Compute rmse_lr
____ = ____

# Print rmse_lr
print('Linear Regression test set RMSE: {:.2f}'.format(rmse_lr))

# Print rmse_dt
print('Regression Tree test set RMSE: {:.2f}'.format(rmse_dt))
コードを編集して実行