カテゴリー
Python

pythonの決定木を調整①

*実際に走らせてないため実行できるかは不明

①サンプル数がどのくらいまで減ったら分岐を止めるか指定する方法

決定木のパラメータの中で、「min_samples_split」がサンプル数がどこまで減ったら分岐を止めるかのやつ。

デフォルトではmin_samples_split=2らしいが、10サンプルになったら止めて欲しいときなんかは、min_samples_split=10にすればいい(はず)

参考URL

model = DecisionTreeRegressor(#この中でmin_samples_split=10を指定)

コメントを残す

WordPress.com で次のようなサイトをデザイン
始めてみよう