1. Học hỏi
  2. /
  3. Khoa Học
  4. /
  5. Pythonで学ぶ木ベースのMachine Learning

Connected

Bài tập

モデルをインスタンス化する

この一連の演習では、回帰木のバイアスとバリアンスの問題を診断します。本演習で定義する回帰木は、auto データセットのすべての特徴量を使って、車の mpg 消費を予測します。

データの前処理は済んでおり、作業環境には特徴量行列 X と配列 y が読み込まれています。さらに、sklearn.tree から DecisionTreeRegressor クラスがインポート済みです。

Hướng dẫn

100 XP
  • sklearn.model_selection から train_test_split をインポートします。
  • データを 70% の学習用と 30% のテスト用に分割します。
  • max_depth を 4、min_samples_leaf を 0.26 に設定して DecisionTreeRegressor をインスタンス化します。