1. 学ぶ
    2. /
  2. コース
    3. /
  3. Pythonで学ぶ木ベースのMachine Learning
Connected

演習

最初の回帰木を学習してみましょう

この演習では、auto-mpg データセットに含まれる6つの特徴量すべてを使って、車の mpg（miles per gallon、燃費）を予測する回帰木を学習します。

データセットはあらかじめ前処理され、80%が学習用、20%がテスト用に分割されています。特徴量行列 X_train と配列 y_train はワークスペースに用意されています。

指示

100 XP
  • sklearn.tree から DecisionTreeRegressor をインポートします。
  • 最大深さを 8、min_samples_leaf を 0.13 に設定して DecisionTreeRegressor のインスタンス dt を作成します。
  • 学習用データに dt をフィットします。